- 要装tensorflow-gpu 先删除cpu版本的tensorflow。
- 版本统一：
- python版本3.6.6
- cuda toolkit 9.0
- cudnn 7
- 可参考 https://www.cnblogs.com/Climbing-Snail/p/9361621.html
- 其中上述链接的第三步，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
- 别的步骤都可以参考以上链接来
- 我代码里用到了CuDNNLSTM (会比LSTM快很多)，有的版本不能用，说一种我实验成功的搭配：python 3.6.6, tensorflow-gpu 1.4.1, cuda 8.0, cudnn 6
