本教程是基于ubuntu 16.04,18.04的安装过程也类似。首先需要去英伟达官网下载CUDA安装包,这里我们选择cuda 10.1,

点进去之后,选择系统版本进行选择,这里我们选择下载run文件,

安装指导里告诉我们,需要用wget命令下载,然后在本地打开一个终端,复制这条代码到终端,运行下载即可。
wget http://developer.download.nvidia.com/compute/cuda/10.1/Prod/local_installers/cuda_10.1.243_418.87.00_linux.run
在这里说一下cuda安装包的名字cuda_10.1.243_418.87.00_linux.run里的418.87意思是当前系统的显卡驱动程序版本不能低于418.87。
下载完成之后,直接在该终端执行
sudo sh cuda_10.1.243_418.87.00_linux.run
接下来就进入安装过程,在该过程中会有一个选项问是否安装nvidia 418.87驱动程序,选择否,其他的默认就可以。
安装过程结束后需要配置环境变量,终端执行
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
然后重新打开一个终端,执行nvcc -V命令,可以看到输出以下信息,说明cuda已经安装成功了。

接下来是安装cudnn,cudnn是给cuda提速的,同样的先去官网下载cudnn,点击Download cuDNN,

然后接下来需要我们登录账号,如果没有账号就得先去注册一个。

登录上账号之后,进入到这个页面,选择对应的版本,在这里选择 Download cuDNN v7.6.5 (November 5th, 2019), for CUDA 10.1,然后开始下载过程,下载下来的应该是一个压缩文件。
下载完成后,解压文件,然后在此处打开终端,cd到该文件夹下,执行以下命令:
cd cuda
sudo cp include/cudnn.h /usr/local/cuda/include/
sudo cp lib64/* /usr/local/cuda/lib64/
sudo chmod a+r /usr/local/cuda/include/cudnn.h
sudo chmod a+r /usr/local/cuda/lib64/libcudnn*
执行完成后,cudnn就安装成功了,然后就可以去跑个程序测试一下。
这篇教程详细介绍了如何在Ubuntu 16.04上安装CUDA 10.1和cuDNN。首先从英伟达官网下载CUDA run文件,确保系统驱动版本不低于418.87。在安装过程中不选择安装驱动,随后配置环境变量。接着下载cuDNN,解压后通过终端命令完成安装。最后,验证CUDA和cuDNN的安装成功。

6469

被折叠的 条评论
为什么被折叠?



