背景:ubuntu14.04 tensorflow-gpu 1.2.0 cuda8.0 cudnn5.1
因为需要用到 tf.bitwise 模块,最低1.3.0支持,所以升级一下下。
注意与tensorflow 1.3.0 配套的 cudnn 是 6.0版本的。
1. 更新tensorflow
A方案: 我是直接在pycharm中更新的,选settings,双击tensorflow-gpu,verify version 1.3.0, install就行。
B方案:
sudo pip uninstall tensorflow-gpu
sudo pip install tensorflow-gpu==1.3
有小伙伴说系统里的 tf 版本可以和pycharm的不一样,这个存疑我还没验证过,平时都用pycharm跑程序。
2. 更新cudnn版本到6.0
在英伟达官网上下载 cudnn6.0 linux版本(需要注册账号,填问卷什么的)
下载完解压 /Downloads/cuda 为解压完的cuda所在路径
cd ~/Downloads/cuda/include
sudo cp *.h /usr/local/cuda/include/
cd ~/Downloads/cuda/lib64
sudo cp libcudnn* /usr/local/cuda/lib64/
sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn*