[tensorflow-gpu][linux][cuda][cudnn][非root] 安装

  1. 要装tensorflow-gpu 先删除cpu版本的tensorflow。
  2. 版本统一:
  • python版本3.6.6
  • cuda toolkit 9.0
  • cudnn 7
  1. 可参考 https://www.cnblogs.com/Climbing-Snail/p/9361621.html
  2. 其中上述链接的第三步,cudnn 下载需要NVIDIA注册,选择cuDNN library,
    下载文件为 cudnn-9.0-linux-x64-v7.3.1.20.solitairetheme8
cp cudnn-9.0-linux-x64-v7.3.1.20.solitairetheme8 cudann-9.0-linux-x64-v7.3.1.20.tgz
tar -xvf cudann-9.0-linux-x64-v7.3.1.20.tgz
[tensorflow-gpu][linux][cuda][cudnn][非root] 安装_第1张图片
image.png
  1. 别的步骤都可以参考以上链接来
  2. 我代码里用到了CuDNNLSTM (会比LSTM快很多),有的版本不能用,说一种我实验成功的搭配:python 3.6.6, tensorflow-gpu 1.4.1, cuda 8.0, cudnn 6

你可能感兴趣的:([tensorflow-gpu][linux][cuda][cudnn][非root] 安装)