安装tensorflow-gpu真的好烦,个人经验来说,最主要的就是找好一下对应关系(我的是python3.5)
1 Linux x86_64 Driver Version与CUDA Toolkit的对应,一定要对应好,CUDA超出了Driver Version的话,是无法使用的
https://developer.nvidia.com/cuda-toolkit-archive
2 CUDA与cuDNN的对应关系
3 在下载cuDNN的时候要注意了,链接
https://developer.nvidia.com/rdp/cudnn-archive
不要看cudnn-(这个数字),应该以v(这个才是版本号).tgz
4 在安装好CUDA和cuDNN之后安装tensorflow,一定要对应好版本号,比如要用1.4版本
$ sudo pip3 install tensorflow-gpu-1.4
可能会下载很慢,需要耐心等待,或者1.4版本可以用tensorflow 实战google深度学习框架中的方法
$ export TF_BINARY_URL=https://storage.googleapis.com/tensorflow/linux/gpu/tensorflow_gpu-1.4.0-cp35-cp35m-linux_x86_64.whl
$ sudo pip3 install –upgrade $TF_BINARY_URL
具体操作步骤,转到
https://blog.ailemon.me/2017/06/06/install-tensorflow-gpu-on-ubuntu-linux/
来看吧