将环境模块加载到Jupyter Notebook /实验室

问题描述

我正在通过服务器运行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命令。因此,首先您需要通过获取模块初始化脚本来定义此Shell函数。可以在类似Red Hat的系统上的/usr/share/Modules/init中找到此类脚本。

运行后:

source /usr/share/Modules/init/bash

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