关于使用gpu跑程序的问题

针对Python3.6的

本人之前一直使用cpu版本的tensorflow,发现神经网络程序运行很慢,于是开始下载gpu版本的tensorflow。查看usr/local/发现有一个cuda-8.0,cuda-9.0文件夹,Ubuntu环境下已经安装好cuda的环境,所以直接下载tensorflow-gpu。

本人下载的是1.4.0版本的与cpu版本的一样,但是运行程序报错了,大概说的是我强行使用gpu,而cpu才是有效的设备(这个大多数人都遇到过),于是去网上求助。在Stack Overflow网站有人遇到相类似的问题,解决方法是卸载两个版本的tensorflow,再重新安装gpu版本的tensorflow,再次运行发现找不到tensorflow了,更头疼了,去网上找攻略,说是没有配置cuda的路径,于是将网上给的路径

export LD_LIBRARY_PATH=LD_LIBRARY_PATH:/usr/local/cuda-9.0/lib64/
export CUDA_HOME=/usr/local/cuda

粘贴到自己家目录下的bashrc里(本人没有管理员权限,不能sudo),保存并在终端source一下,输入source  /home/用户名/bashrc

然后继续运行程序,发现说libcudnn.so.6.0找不到,网上说是下载的tensorflow-gpu=1.4.0需要对应的cuda是版本6的,而我的是8和9,这时候我卸载gpu版的tensorflow,找了网上说对应cuda-9.0版本的tensorflow-gpu, 重新下了一个1.9.0的。发现程序可以运行了。

再不行,就全部卸了(卸干净,所有的关于tensor),用conda install tensorflow-gpu==1.x.x 

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

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值