时间戳

首页 > 解决方案 > 将环境模块加载到 Jupyter Notebook/Lab

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

问题描述

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

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

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

有没有办法我可以使用module load或正确指向包的位置，因为不幸的是我正在使用的 Python 库在尝试导入它时一直抛出这个错误

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

所有帮助表示赞赏

标签： pythonjupytercluster-computingcondaenvironment-modules

解决方案

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

运行后：

source /usr/share/Modules/init/bash

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

推荐阅读