一、装置驱动

1. 下载驱动

NVDIA driver search page搜寻你的显卡须要的驱动型号并下载(如图)。

搜寻,而后download

2. 禁用nouveau

关上终端,输出:

sudo gedit /etc/modprobe.d/blacklist.conf 

在blacklist.conf文件开端加上这两行,并保留:

blacklist nouveau

而后执行命令:

sudo update-initramfs -u  //利用更改

重启电脑,验证是否禁用nouveau,这一条是用来禁用nouveau驱动,之后也不须要改回来。:

lsmod | grep nouveau

没有信息返回阐明已禁用了如图:

3. 装置NVIDIA驱动

关上terminal卸载旧版本NVIDIA驱动:

sudo apt-get remove --purge nvidia*

进入到下载好的.run文件夹下给驱动run文件赋予执行权限:

sudo chmod  a+x NVIDIA-Linux-x86_64-460.84.run //对应本人下载的驱动名称

装置NVIDIA驱动

sudo ./NVIDIA-Linux-x86_64-460.84.run -no-x-check -no-nouveau-check -no-opengl-files// 留神:前面三个选项的后面都是:减号“-”
  • -no-x-check:装置驱动时敞开X服务
  • -no-nouveau-check:装置驱动时禁用nouveau
  • -no-opengl-files:只装置驱动文件,不装置OpenGL文件 (最重要)

装置过程中的选项:

The distribution-provided pre-install script failed! Are you sure you want to continue? 抉择 yes 持续。Would you like to register the kernel module souces with DKMS? This will allow DKMS to automatically build a new module, if you install a different kernel later?  抉择 No 持续。问题没记住,选项是:install without signing问题大略是:Nvidia's 32-bit compatibility libraries? 抉择 No 持续。Would you like to run the nvidia-xconfigutility to automatically update your x configuration so that the NVIDIA x driver will be used when you restart x? Any pre-existing x confile will be backed up.  抉择 Yes

这些选项如果抉择谬误可能会导致装置失败,没关系,只有后面不出错,多尝试几次就好。

查看NVIDIA版本测验是否装置实现

二、装置CUDA

1. 下载

去官方网站下载合乎本人电脑的cuda版本,抉择好相应的版本后依照上面提供的形式进行装置。

最初一项Installer Type倡议抉择runfile [local],因为命令行少,更不便。

以这个图为例,就是先下载cuda_11.4.0_470.42.01_linux.run这个文件(能够间接把链接放到浏览器去下载)。

2. 装置

下载之后,运行如下命令

sudo sh cuda_11.4.0_470.42.01_linux.run

输出accept回车

这个中央不要下载Driver,因为之前曾经装置完了,肯定要抉择CUDA Toolkit 10.2。

勾销抉择的办法是:光标停留在Driver那一行上,而后回车,使[]里的X隐没

抉择Install回车

看到这页面预计是装置完了,输出以下命令验证是否胜利:

cat /usr/local/cuda/version.txt

如果显示没有这个文件,就到该文件夹上来看看有没有一个version.txt,外面如果有version.json,且json中有版本信息,也能够。

接下来设置环境变量

gedit ~/.bashrc

退出如下环境变量(留神批改为本人的门路):

export PATH=/usr/local/cuda-11.4/bin${PATH:+:${PATH}}export LD_LIBRARY_PATH=/usr/local/cuda-11.4/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}

保留退出,终端运行:

source ~/.bashrc

终端运行:

nvcc -V

如果显示上面的文子就阐明装置胜利了。

三、装置cuDNN

进入官网(须要注册)

注册之后进入,批准协定,显示如下

抉择本人的版本,我须要CUDA11,就点第一个 。

当初有两种装置cuDNN的形式,第一种形式能够测验;第二种形式要下载的文件比拟少。

第一种:Debian File模式的装置

1. 下载

抉择ubuntu20的这个三文件下载到本地,

2. 装置

而后执行以下三个命令

sudo dpkg -i libcudnn8_8.2.2.26-1+cuda11.4_amd64.debsudo dpkg -i libcudnn8-dev_8.2.2.26-1+cuda11.4_amd64.debsudo dpkg -i libcudnn8-samples_8.2.2.26-1+cuda11.4_amd64.deb

3. 验证

当抉择Debian File进行装置时会在/usr/src/cudnn_samples_v8有一些cudnn的例子

在任意目录下开展终端,运行以下命令,通过编译mnistCUDNN sample进行验证

cp -r /usr/src/cudnn_samples_v8/ $HOMEcd  $HOME/cudnn_samples_v8/mnistCUDNNmake clean && make./mnistCUDNN

如果在执行sudo make”时报以下编译谬误 :fatal error: FreeImage.h

mnistCUDNN  sudo makeCUDA_VERSION is 11010Linking agains cublasLt = trueCUDA VERSION: 11010TARGET ARCH: x86_64HOST_ARCH: x86_64TARGET OS: linuxSMS: 35 50 53 60 61 62 70 72 75 80 86test.c:1:10: fatal error: FreeImage.h: No such file or directory    1 | #include "FreeImage.h"      |          ^~~~~~~~~~~~~compilation terminated.

则执行:sudo apt-get install libfreeimage3 libfreeimage-dev,而后从新验证。

如果装置胜利将会有如下图所示的:Test passed!

这种验证会在home目录下生成一个cudnn_samples_v8文件加,验证胜利后删除即可。

第二种:Tar File模式的装置

1. 下载

抉择ubuntu20的这个文件下载到本地,

2. 装置

下载的是tgz的压缩包,须要先解压:

tar -xzvf cudnn-11.4-linux-x64-v8.2.2.26.tgz

执行装置,其实就是拷贝头文件和库文件并给予权限

sudo cp cuda/include/cudnn.h /usr/local/cuda/includesudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64sudo chmod a+r /usr/local/cuda/include/cudnn.h sudo chmod a+r /usr/local/cuda/lib64/libcudnn*

实现。解压操作会在当前目录生成一个cuda文件夹,删除即可。

参考(包含局部图):

https://www.zhihu.com/collect...

https://blog.csdn.net/dudu815...

https://blog.csdn.net/xhw205/...