ubuntu18.04.4LTS系统,安装cuda10.1与cudnn

tech2025-05-21  1

目录

一、安装cuda1、下载安装包2、安装3、检测 二、安装cudnn1、 下载cudnn压缩包2、安装3、检测cudnn安装是否成功

一、安装cuda

1、下载安装包

首先去官网下载cuda安装包, 链接.

我找了半天cuda10.1,没找到,其实这样找:

1、进去之后,直接点击下载 2、点击下图所示:

3、下图中选择一个 4、选择deb(local)安装类型,然后下载。

2、安装

1、进入安装包所在的目录:

cd 安装包所在的目录

2、运行.deb文件,其实就是解压

sudo dpkg -i 你的.deb文件

第一次应该会让你安装一个类似: sudo apt-key add /var/cuda-repo-/7fa2af80.pub这样的包,安装即可。然后再次运行上面的命令。

3、更新一下下载源:

sudo apt-get update

4、安装cuda

sudo apt-get install cuda

5、添加环境变量

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

这里注意,至于是不是cuda-10.1,去/usr/local/下看看,有的也可能不是。

最后使其生效:

source ~/.bashrc

6、重启

reboot

3、检测

命令行输入:

nvidia-smi

显示下面这样,说明驱动没问题。 输入下面命令,若显示下图所示,说明安装cuda成功。

nvcc -V

查看安装的cuda版本:

cat /usr/local/cuda/version.txt

二、安装cudnn

1、 下载cudnn压缩包

去cudnn的官网下载对应的版本:官网

普通机器 :x86_64architecture 服务器机器 :Power architecture

选择 cuDNN Library for linux

按上图下载下来的是压缩包。要是选择其他的,应该是.deb文件,那就是另外的一种安装方式,我没试过,所以这里也就没写。

2、安装

将下载下来的压缩包解压(记得解压到home下,右键提取就是解压),当然也可以使用命令来解压:

sudo tar -xvzf cudnn-10.1-linux-x64-v7.6.5.32

解压到home下,可以看到cuda文件夹,在当前home目录打开终端,执行下面的命令:

sudo cp cuda/include/cudnn.h /usr/local/cuda/include/ 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*

这就已经完成了安装。其实cudnn的安装,就是拷贝头文件和库文件到cuda 对应的文件夹下,并给予权限。

3、检测cudnn安装是否成功

查看cudnn的版本信息,在终端输入:

cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2

最新回复(0)