1. 卸载原来的cuda:
sudo apt-get remove cuda
sudo apt-get autoclean
sudo apt-get autoremove
sudo apt-get remove cuda*
注意在system setting-software &update里可能有原来cuda的更新源,也需要删掉。
2. 下载cuda8:
官网:https://developer.nvidia.com/cuda-toolkit-archive
选择版本CUDA Toolkit 8.0 GA2 (Feb 2017)
Linux-x86_64-ubuntu-16.04-deb(local)
4. 安装
sudo dpkg -i cuda-repo-ubuntu1604-8-0-local-ga2_8.0.61-1_amd64.deb
sudo apt-get update
sudo apt-get install cuda-8-0
5. 下载cudnn
https://developer.nvidia.com/rdp/cudnn-download
注册登录下载,但是经常登录不来,这里有一个百度云下载:
链接: https://pan.baidu.com/s/1Os1tk16JB1uH9wXz_9xbMQ 提取码: 7grq
6. 解压: tar -xzvf cudnn-8.0-linux-x64-v5.1.tgz # 用5.0好像后面运行tennsorflow报错
# 拷贝文件
sudo cp cuda/include/cudnn.h /usr/local/cuda/include
sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64
sudo chmod a+r /usr/local/cuda/lib64/libcudnn*
# 查看是否安装成功
nvcc -V