解决 cuda10 环境上 tf.test.is_gpu_available() 为false

.升级Tensorflow 1.15 以及 cuda10,进行分布式训练 ,发现gpu  内存都只占用 145M,开始测试单个卡也是145M ,使用如下简单测试下gpu是否能调用

import tensorflow as tf
import os
os.environ['CUDA_VISIBLE_DEVICES'] = "1"

print(tf.test.is_gpu_available())

为false

cat /usr/local/cuda/version.txt   输出正常

配置下cuda 环境变量   vim   ~/.bashrc  

export CUDA_HOME=/usr/local/cuda
export PATH=$PATH:$CUDA_HOME/bin
export LD_LIBRARY_PATH=/usr/local/cuda-10.0/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}

source   ~/.bashrc    jupyter 中当前Terminal   测试 后为True ,但新开一个Terminal或在 ipython中 运行   仍为False

vim  /etc/profile  如上配置    再运行 ldconfig 后 Gpu 终于正常 调用

 参考 https://www.cnblogs.com/sddai/p/11135941.html

评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值