关于使用gpu跑程序的问题

本人之前一直使用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的。发现程序可以运行了。

猜你喜欢

转载自blog.csdn.net/qq_33772192/article/details/86234959