[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
image.png
  1. 别的步骤都可以参考以上链接来
  2. 我代码里用到了CuDNNLSTM (会比LSTM快很多)，有的版本不能用，说一种我实验成功的搭配：python 3.6.6, tensorflow-gpu 1.4.1, cuda 8.0, cudnn 6
最后编辑于 ：
©著作权归作者所有,转载或内容合作请联系作者
平台声明：文章内容（如有图片或视频亦包括在内）由作者上传并发布，文章内容仅代表作者本人观点，简书系信息发布平台，仅提供信息存储服务。

推荐阅读更多精彩内容