ssh远程使用服务器GPU,tf.test.is_gpu_available()的运行结果为false

远程服务器环境:

ubuntu18.04
python3.6
tensorflow-gpu 1.14
cudnn 7.4.1
cuda 10.0

客户端在pycharm上ssh远程连接服务器。
在远程服务器上新建的虚拟环境上安装如上包,因为是非root用户,安装过程参考的是:

https://blog.csdn.net/hizengbiao/article/details/88625044?spm=1001.2014.3001.5502

配置的用户环境变量

export PATH=/home/name/miniconda3/envs/envname/cuda-10.0/bin:$PATH
export LD_LIBRARY_PATH=/home/name/miniconda3/envs/envname/cuda-10.0/lib64

问题

安装完tensorflow-gpu版本后,在客户端pycharm的terminal上使用tf.test.is_gpu_available(),测试是否安装成功,输出true。但是同样在客户端的.py文件上测试输出false。

解决方法

1.如果terminal上tf.test.is_gpu_available()测试就是false的话,应该是版本适配问题,请务必在安装时就选择好各个适配版本适配版本
2.如果和我一样,terminal上测试true,.py文件又为false
参考解决方法:

https://blog.csdn.net/vivohchj/article/details/107673281
pycharm没有找到CUDA的支持文件,还记得安装CUDA时我们在.bashrc文件里写入过两个环境变量吗,问题就在LD_LIBRARY_PATH,环境变量里明明有,但是pycharm就是读不到。

使用解决方法1,为每个.py文件添加了环境变量
"RUN"下面的"Edit Configuration"中,environment一栏可以看到pycharm获取的环境变量,发现确实没有LD_LIBRARY_PATH。
在这里插入图片描述
添加进去
在这里插入图片描述
输出true

解决方法2暂时还没有尝试

  • 1
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值