1.安装显卡驱动
简单一步在ubuntu18.04下nvidia driver的安装
直接通过Ubuntu自带UI界面安装,应用更改自动安装,重启生效
卸载CUDA和cuDNN看这
2.安装CUDA10.1
官网下载地址
运行.run文件
sudo sh cuda_10.1.105_418.39_linux.run
- 1
因为已经安装了NVIDIA 显卡驱动,所以这里不勾选安装驱动
等待片刻,出现安装成功
添加环境变量
sudo vim ~/.bashrc
- 1
export CUDA_HOME=/usr/local/cuda-10.1
export PATH=/usr/local/cuda-10.1/bin${PATH:+:$PATH}
export LD_LIBRARY_PATH=/usr/local/cuda-10.1/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
- 1
- 2
- 3
验证
nvcc --version
- 1
3.安装cudnn7.6.2
官网下载地址
下载cuDNN Library for Linux
,解压,在解压的文件夹中打开终端
复制文件到目录
sudo cp include/cudnn.h /usr/local/cuda/include/
sudo cp lib64/lib* /usr/local/cuda/lib64/
- 1
- 2
建立软链接
在/usr/local/cuda/lib64目录下打开终端,执行如下指令:
sudo chmod +r libcudnn.so.7.6.2
sudo ln -sf libcudnn.so.7.6.2 libcudnn.so.7
sudo ln -sf libcudnn.so.7 libcudnn.so
sudo ldconfig
- 1
- 2
- 3
- 4
查看当前cudnn的版本
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
- 1