将环境模块加载到 Jupyter Notebook/Lab

Load Environment Modules into Jupyter Notebook/Lab

我是 运行 JupyterLab 通过服务器(远程管理,我是没有 sudo 访问权限的用户)。我想使用 CUDA,但为了这样做,我必须在终端中使用 module load bash 命令将其作为模块加载。

CUDA 包已经安装,所以我假设模块加载只是添加指向包所在位置的路径。我无法在 Jupyter 中使用 !module load cuda,所以我根本无法使用 CUDA。 CUDA 安装也需要 sudo 权限。

我试过使用 %env 添加环境变量,但除此之外我真的卡住了。

有什么方法可以使用 module load 或正确指向包所在的位置,因为不幸的是,我正在使用的 Python 库在尝试导入它时一直抛出此错误

libcudart.so.9.2: cannot open shared object file: No such file or directory

感谢所有帮助

似乎 module 命令在您尝试使用的环境中无法识别。所以首先你需要通过 sourcing 模块初始化脚本来定义这个 shell 函数。这样的脚本可以在类似 Red Hat 的系统上的 /usr/share/Modules/init 中找到。

运行之后:

source /usr/share/Modules/init/bash

您应该定义 module shell 函数,然后执行 module avail 将告诉您 cuda 模块文件是否可用于加载 module load cuda.