我的系统是Ubuntu18.04 64位,我选择下载最新版本的CUDA10.1:
![](http://www.linuxchn.com/wp-content/uploads/2021/04/2021042716573283-1024x564.png)
输入以下命令安装CUDA
sudo sh cuda_10.1.105_418.39_linux.run
显卡驱动已在上面安装了,这里一定要选择不安装驱动,如下
![](http://www.linuxchn.com/wp-content/uploads/2021/04/2021042716575453.jpg)
![](http://www.linuxchn.com/wp-content/uploads/2021/04/2021042716581129.png)
安装完后会有警告提示信息,忽略就行。是因为在安装CUDA时没有选择安装显卡驱动。
![](http://www.linuxchn.com/wp-content/uploads/2021/04/2021042716584384.png)
安装完后,在.bashrc文件末尾添加环境变量
sudo gedit ~/.bashrc
export PATH="/usr/local/cuda-10.1/bin:$PATH"
export LD_LIBRARY_PATH="/usr/local/cuda-10.1/lib64:$LD_LIBRARY_PATH"
保存退出后,输入以下命名
source ~/.bashrc
输入以下命令,测试是否安装成功
cd /usr/local/cuda/samples/1_Utilities/deviceQuery
sudo make
./deviceQuery
如 Result = PASS表示CUDA安装成功。
安装cuDNN7.5
在官网下载cuDNN,选择自己对应的系统和已安装的CUDA版本下载cuDNN。(要注册才能下载)
我选择的是cuDNN v7.5.1 (April 22, 2019), for CUDA 10.1 for Linux
![](http://www.linuxchn.com/wp-content/uploads/2021/04/2021042717025947.png)
下载完成后,输入以下命令解压文件
tar -zxvf cudnn-10.1-linux-x64-v7.5.1.10.tgz
输入以下命名,将cudnn相关文件拷贝到已安装的cuda文件中
sudo cp cuda/include/cudnn.h /usr/local/cuda/include/ (#解压后的文件夹名为cuda)
sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64/
sudo chmod a+r /usr/local/cuda/include/cudnn.h (#增加所有用户对文件的可执行权限)
sudo chmod a+r /usr/local/cuda/lib64/libcudnn*