screen或者tmux中因CUDA无法调用GPU问题

微凉离别意 10天前   阅读数 8 0

说明:

screen 与 tmux 都是可以实现远程离线训练模型的工具,
可以使训练命令在服务器后台运行,当前ssh断掉连接也不会影响训练指令.

但是最近发现若是在screen中使用虚拟环境,tensorflow占用了GPU但是未使用GPU计算,
并且若是使用theano可能会报如下错误:

 You forced the use of gpu device gpu,but CUDA initialization failed with error:
 cuda unavailable

原因:

  会话环境可能改变了环境变量 LD_LIBRARY_PATH

解决方案:

先在screen会话外的命令行中确认能运行的环境的LD_LIBRARY_PATH是多少
即:   echo $LD_LIBRARY_PATH

然后进入screen会话中定义该环境变量,PATH为上面echo输出的正常变量地址
即:   export LD_LIBRARY_PATH="PATH"
并且虚拟环境名称与会话名称最好不同

注意:本文归作者所有,未经作者允许,不得转载

全部评论: 0

    我有话说: