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

  • Post author:
  • Post category:linux


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

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


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

CUDA与Driver Version对应关系

2 CUDA与cuDNN的对应关系

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/

来看吧



版权声明:本文为Junizxr原创文章,遵循 CC 4.0 BY-SA 版权协议,转载请附上原文出处链接和本声明。