Jupyterhub找不到GPU问题

10 篇文章 0 订阅

非 root 用户安装 cuda/cudnn 环境后,通过修改 PATH 和 LD_LIBRARY_PATH 变量,可以使 tensorflow 成功识别调用 GPU。但是,运行 jupyterhub notebook 后发现 GPU 调用失败... 

  •  首先,在 jupyter notebook 环境中运行以下命令,可以查看 tensorflow 识别出的硬件。 
import tensorflow as tf
tf.config.list_physical_devices()
  • 看到输出结果中并未包含 GPU。但是在 terminal 环境的 Python 中运行相同的命令,看到的结果是包含 GPU 的。
  • 刚开始以为是 cuda 没安装正确,重新安装了一下,依然无法识别。接着运行以下命令查看 Python 识别到的系统环境变量,看到其中没有了自定义的 LD_LIBRARY_PATH。
import os
print(os.environ)
  • 初步猜测是 Jupyter notebook 运行是忽略了 ~/.zshrc 中的配置。
  • 但是,可以看到 PATH 变量显示的是修改过的内容。
    因此打算生成 jupyter 的配置文件(命令如下)后,强制指定 LD_LIBRARY_PATH。
jupyterhub notebook --generate-config
  • 推测原因出在这里,single -user 的 JupyterHub 会隐藏掉系统的环境变量,只留下 whitelist 中的这几个。
  • 因此把 LD_LIBRARY_PATH 加入 .env_keep 列表
c.Spawner.env_keep = ['PATH', 'PYTHONPATH', 'CONDA_ROOT', 'CONDA_DEFAULT_ENV', 'VIRTUAL_ENV', 'LANG', 'LC_ALL', "LD_LIBRARY_PATH"]

  • 0
    点赞
  • 4
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值