Ubuntu下安装cuda

来源:互联网 发布:淘宝升级 编辑:程序博客网 时间:2024/05/20 05:09

1、官网下载需要的版本https://developer.nvidia.com/accelerated-computing-toolkit

2、选择Ubuntu下的deb包 local版本

3、cuda9.0安装指令:

  1. `sudo dpkg -i cuda-repo-ubuntu1604-9-0-local_9.0.176-1_amd64.deb`
  2. `sudo apt-key add /var/cuda-repo-<version>/7fa2af80.pub`
  3. `sudo apt-get update`
  4. `sudo apt-get install cuda-9-0`
4、cuda8.0安装指令:

  1. `sudo dpkg -i cuda-repo-ubuntu1604-8-0-local-ga2_8.0.61-1_amd64.deb`
  2. `sudo apt-get update`
  3. `sudo apt-get install cuda-8-0`
上面的指令相比官网提供的参考安装教程,主要的不同点在于安装cuda 的指令上带上了明确的版本!这一点对于安装过多个cuda版本的机子非常受用,且可以减少很多地方。


附加:卸载指令

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,使上述设置立即生效。