一、准备工作
Nvidia驱动的安装,看我另外一个文章ubuntu18.04 安装Nvidia驱动的三种方式(必看)_低吟浅笑-CSDN博客
打开终端,输入
nvidia-smi
出来下列信息:
出来该信息,说明驱动无误。查看驱动显示出来的CUDA版本,所安装的版本尽量与这个接近,驱动可以向下兼容,也就是可以安装比显示出来的CUDA版本低的版本。如图所示
二、安装CUDA
1. 官网下载相应的版本,需要考虑nvidia驱动:CUDA Toolkit Archive | NVIDIA Developer
2. 根据 nvidia驱动的版本下载cuda,注意下面有个 wget 网址的选项,如果wget下载较慢,可以打开链接直接下载。
3. 安装 cuda
进入下载好的文件夹,输入
sudo sh cuda_10.0.130_410.48_linux.run
注意,新版的cuda安装是个小窗口,需要按 Enter 选择哪个不安装,旧版是通过提问的方式来的。
第一个选项选 accept
除了安装 驱动 的选项,其他都选 yes
4.配置环境变量
输入下面的命令打开bashrc
sudo vi ~/.bashrc
或者
sudo gedit ~/.bashrc
在最下面添加下面内容,注意 里面的 cuda-10.0 需要换成自己的版本,自己可以去 /usr/local/ 下面看一下呢
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-10.0/lib64
export PATH=$PATH:/usr/local/cuda-10.0/bin
export CUDA_HOME=$CUDA_HOME:/usr/local/cuda-10.0
重新启动配置文件
source ~/.bashrc
三、安装CUDNN