准备在实验室电脑上使用tensorflow仿真。由于之前已经安过cuda7.5,现在不够用了要升级,一步步来。
1.升级cuda7.5到cuda8.0,安装cudnn5.1
首先卸载cuda7.5
cd /usr/local/cuda/bin
sudo ./uninstall_cuda_7.5.pl
在System Settings->Software & Updates->Additional Drivers把显卡的最新的闭源驱动装了,重启之后正常的话,显卡驱动就是装好了。cuda 8.0下载之后是一个.run文件,给它用chmod +x添加可执行属性之后用sudo运行,别安装里面的驱动,只装cuda tookits。下载cudnn,要注册一个账号,看清楚了Tensorflow目前最新版本是1.1.0-rc0,需求cuda 8.0和cudnn 5.1 (5.0不行)。cudnn 5.1下载以后解压可以看到cuda文件夹,按照下面命令把相应头文件和库文件拷贝到系统路径下面就行了。
cd cuda
sudo cp include/*.h /usr/local/cuda/include/ (注意这里是你自己的cuda的安装地址,复制*.h文件)
sudo cp lib64/libcudnn* /usr/local/cuda/lib64/
sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn*
2.使用pip安装tensorflow
pip install tensorflow_gpu-1.1.0rc2-cp27-none-linux_x86_64.whl