Ubuntu 18.04安装Cuda和Cudnn
软件配置
- Ubuntu 18.04 LTS
- Cuda 10.1
- Cudnn 7
step 1: 安装Nvidia显卡驱动
-
使用
ubuntu-drivers devices
查看, 安装界面中推荐的驱动版本,例如:driver : nvidia-driver-440 - third-party free recommanded
. -
卸载旧的Nvidia驱动
sudo apt remove --purge nvidia*
-
更新显卡驱动软件源
sudo apt update sudo apt upgrade sudo add-apt-repository ppa:graphics-drivers
-
安装推荐的显卡驱动
sudo apt install nvidia-driver-440
-
测试: 终端输入
nvidia-smi
,显示显卡信息即为安装成功。
step 2: 安装 Cuda 10.1
-
官网Nvidia 下载 cuda_10.1.168_418.67_linux.run 文件。
-
安装Cuda.
sudo sh cuda_10.1.168_418.67_linux.run
安装过程中会出现一个选择界面,第一项显卡驱动不选,之后默认安装即可。
. -
安装成功
-
环境变量设置
sudo vim ~/.bashrc export PATH="/usr/local/cuda-10.1/bin:$PATH" export LD_LIBRARY_PATH="/usr/lcoal/cud10.1/lib64:$LD_LIBRARY_PATH" source ~/.bashrc
-
测试安装是否成功
cd /usr/local/cuda-10.1/samples/1_Utilities/deviceQuery sudo make ./deviceQuery
出现Result = Pass则表示安装成功
Step 3: 安装cuDNN
-
解压cuDNN到相应的目录,运行下面命令:
sudo cp cuda/include/cudnn.h /usr/local/cuda-10.1/include sudo cp cuda/lib64/libcudnn* /usr/local/cuda-10.1/lib64 sudo chmod a+r /usr/local/cuda-10.1/include/cudnn.h sudo chmod a+r /usr/local/cuda-10.1/lib64/libcudnn*
-
下载
libcudnn7_7.6.5.32-1+cuda10.1_amd64.deb, libcudnn7-dev_7.6.5.32-1+cuda10.1_amd64.deb, libcudnn7-doc_7.6.5.32-1+cuda10.1_amd64.deb
.sudo dpkg -i libcudnn7_7.6.5.32-1+cuda10.1_amd64.deb sudo dpkg -i libcudnn7-dev_7.6.5.32-1+cuda10.1_amd64.deb sudo dpkg -i libcudnn7-doc_7.6.5.32-1+cuda10.1_amd64.deb
在这个过程中可能会出现 libcudnn.so.7 is not a symbolic link 的问题,产生该问题的原因是该文件是一个软连接的文件,但是没找到具体连接的实体。解决方法就是在 /usr/local/cuda-xx/lib64 中查找 libcudnn.so.7.xx的文件,将其连接在一起:
sudo ln -sf /usr/local/cuda-xx/lib64/libcudnn.so.7.xx /usr/local/cuda-xx/lib64/libcudnn.so.7
-
检测是否安装成功
cp -r /usr/local/cudnn_samples_v7 ~/ cd ~/cudnn_samples_v7 make clean && make ./mnistCUDNN
出现 Test passed 即为安装成功。