Linux系统下安装TensorFlow的GPU版本(亲测可行)

安装tensorflow-gpu真的好烦,个人经验来说,最主要的就是找好一下对应关系(我的是python3.5)

1 Linux x86_64 Driver Version与CUDA Toolkit的对应,一定要对应好,CUDA超出了Driver Version的话,是无法使用的

https://developer.nvidia.com/cuda-toolkit-archive

Linux系统下安装TensorFlow的GPU版本(亲测可行)_第1张图片

2 CUDA与cuDNN的对应关系

Linux系统下安装TensorFlow的GPU版本(亲测可行)_第2张图片

 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/ 来看吧

 

你可能感兴趣的:(Linux系统下安装TensorFlow的GPU版本(亲测可行))