是否可以从 jupyter 笔记本单元内的管道设置 env？

Question

在 jupyter 笔记本单元格中运行此命令

!whereis cudnn.h

得到这个输出

cudnn: /usr/include/cudnn.h

在 jupyter 笔记本单元格中运行此命令

%env CUDNN_H_PATH=/usr/include/cudnn.h

设置环境变量

在 jupyter 笔记本单元格中运行此命令

%env CUDNN_H_PATH=$(whereis cudnn.h)

得到这个

!echo $CUDNN_H_PATH
$(whereis cudnn.h)

问题

是否可以将!whereis cudnn.h的输出设置为环境变量CUDNN_H_PATH？

Answer 1

有可能：存储!whereis cudnn.h的结果，这将是一个列表，并获得第一项。然后只需将其传递给%env：

path = !whereis ls # using ls instead of cudnn.h, but would work either way
path = path[0]
%env CUDNN_H_PATH=$path

!echo $CUDNN_H_PATH
ls: /bin/ls

但是，如果您只想获取来自!whereis cudnn.h 的路径（没有cudnn:），您可以执行以下操作：

path = !whereis ls
path = path[0].split(' ')[1]
%env CUDNN_H_PATH=$path

!echo $CUDNN_H_PATH
/bin/ls

如果!whereis cudnn.h 没有找到您要查找的内容，您必须注意IndexError。