1、官网下载需要的版本https://developer.nvidia.com/accelerated-computing-toolkit
2、选择Ubuntu下的deb包 local版本
3、cuda9.0安装指令:
- `sudo dpkg -i cuda-repo-ubuntu1604-9-0-local_9.0.176-1_amd64.deb`
- `sudo apt-key add /var/cuda-repo-<version>/7fa2af80.pub`
- `sudo apt-get update`
- `sudo apt-get install cuda-9-0`
- `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`
附加:卸载指令
sudo apt-get remove cuda(如果系统有多个cuda的话这里就要带上版本号)
sudo apt-get autoremove
5、配置环境变量以及拷贝cudnn文件
最后需要配置环境变量,让我们安装的CUDA生效。
(1)打开sudo gedit ~/.bashrc,在最后面添加两行如下:
export PATH=/usr/local/cuda-8.0/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-8.0/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
(2)设置环境变量和动态链接库,打开sudo gedit /etc/profile,在文件末尾加入:
export PATH=/usr/local/cuda/bin:$PATH
(3)创建链接文件,打开sudo gedit /etc/ld.so.conf.d/cuda.conf,在文件中添加:
/usr/local/cuda/lib64
最后执行 sudo ldconfig,使上述设置立即生效。