安装平台及环境
CPU:i9-9900k 桌面级
GPU:RTX 2080 移动版
系统: Ubuntu 18.04.3 LTS
1, 在安装 CUDA 之前确保环境满足安装条件
2, 进入 NVIDIA 官网下载适合自己机器的 CUDA 版本, 官网下载, 如图所示, 按照 Installation Instructions 来进行,
- wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/cuda-ubuntu1804.pin
- sudo mv cuda-ubuntu1804.pin /etc/apt/preferences.d/cuda-repository-pin-600
- wget http://developer.download.nvidia.com/compute/cuda/10.1/Prod/local_installers/cuda-repo-ubuntu1804-10-1-local-10.1.243-418.87.00_1.0-1_amd64.deb
- sudo dpkg -i cuda-repo-ubuntu1804-10-1-local-10.1.243-418.87.00_1.0-1_amd64.deb
- sudo apt-key add /var/cuda-repo-10-1-local-10.1.243-418.87.00/7fa2af80.pub
- sudo apt-get update
- sudo apt-get -y install cuda
3, 安装完成后, 添加环境变量, 打开 bashrc 文件,
命令行输入: sudo gedit ~/.bashrc , 然后在文件最后添加下面 3 行, 保存
- export CUDA_HOME=/usr/local/cuda
- export PATH=$PATH:$CUDA_HOME/bin
- export LD_LIBRARY_PATH=/usr/local/cuda-10.1/lib64${
- LD_LIBRARY_PATH:+:${
- LD_LIBRARY_PATH
- }
- }
然后刷新下环境变量,
输入: source ~/.bashrc
4, 测试下 CUDA 是否安装成功,
方法 1: 输入 nvcc -V, 显示如下, 说明没问题
方法 2: 输入如下, 显示如图说明安装成功
- cd /usr/local/cuda/samples/1_Utilities/deviceQuery
- sudo make
- ./deviceQuery
5, 安装 cuDNN 7.6.5 版本, 转到官网下载, 下载前先注册一下, 填个调查问卷, 根据自己的环境和架构选择包, 下载到本地
切换到下载目录 cd Downloads , 然后按照以下操作进行,
解压 cuDNN Library for Linux, 输入: tar -zxvf cudnn-10.1-Linux-x64-v7.6.5.32.tgz
将解压出来的文件复制到安装的 CUDA 环境中, 输入: sudo cp cuda/include/cudnn.h /usr/local/cuda/inlude
sudo cp cuda/lib64/libcudnn*/usr/local/cuda/lib64
更改权限, 输入: sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn*
安装 Deb 包, cuDNN Runtime Library for Ubuntu18.04(Deb),cuDNN Developer Library for Ubuntu18.04(Deb),cuDNN Code Samples and User Guide for Ubuntu18.04(Deb)
分别输入: sudo dpkg -i libcudnn7_7.6.5.32-1+cuda10.1_amd64.deb
- sudo dpkg -ilibcudnn7-dev_7.6.5.32-1+cuda10.1_amd64.deb
- sudo dpkg -i libcudnn7-doc_7.6.5.32-1+cuda10.1_amd64.deb
安装结束后, 测试一下,
方法 1, 输入: cp -r /usr/src/cudnn_samples_v7/ ~
- cd ~/cudnn_samples_v7/mnistCUDNN
- make clean && make
- ./mnistCUDNN
出现 Test passed! 没有报错即安装成功
方法 2, 输入: cd /usr/local/cuda/samples/1_Utilities/deviceQuery
- sudo make
- ./deviceQuery
出现 PASS 结果, 说明安装成功.
来源: http://www.bubuko.com/infodetail-3283297.html