解决docker容器使用ssh连接时,无法调用GPU问题

tensorflow 同时被 2 个专栏收录
10 篇文章 2 订阅
1 篇文章 0 订阅

问题现象

直接使用docker run命令运行tensorflow程序正常
但是用pycharm远程调试时上述容器时运行tensorflow程序失败,提示CUDA错误

问题原因

pycharm远程调试使用的是ssh连接,经测试用ssh远程连接运行也失败
对比docker run和ssh连接两种情况下的环境变量,发现ssh连接少了很多环境变量

解决方案

临时使用环境变量

export $(cat /proc/1/environ |tr '\0' '\n' | xargs -0)

永久使用环境变量

在/etc/profile文件中追加export $(cat /proc/1/environ |tr '\0' '\n' | xargs -0)

参考

https://xuxinkun.github.io/2019/03/12/docker-env/

  • 0
    点赞
  • 0
    评论
  • 0
    收藏
  • 一键三连
    一键三连
  • 扫一扫,分享海报

©️2021 CSDN 皮肤主题: 技术黑板 设计师:CSDN官方博客 返回首页
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值