龙空技术网

WSL2 Ubuntu安装NVIDIA cuDNN深度学习库

一笑看乾坤 47

前言:

眼前朋友们对“ubuntu的库”大约比较关怀,咱们都需要剖析一些“ubuntu的库”的相关资讯。那么小编在网上汇集了一些有关“ubuntu的库””的相关知识,希望姐妹们能喜欢,朋友们一起来学习一下吧!

WSL2 Ubuntu安装NVIDIA cuDNN深度学习库,有好几种方法:

1、有下载tar包安装法

2、下载本地系统安装包安装,例如:

sudo dpkg -i cudnn-local-repo-$distro-8.x.x.x_1.0-1_amd64.deb

3、系统包管理器安装 ,即 apt install xxx

查看我们cudu版本号,运行如下命令

nvidia-sm
export cuda_version=11.6export cudnn_version=8.5.0

去下面这个网站,看到8.5.0支持ubuntu22.04,那我么设置cudnn版本就为8.5.0

cuda的版本是11的我们用第2种方法安装:

如果你的cuda是12以上的版本,则可以用以下命令安装 :

wget 
sudo dpkg -i cuda-keyring_1.1-1_all.deb
sudo apt search cudnn

安装 cuDNN library 和 cuDNN 示例,示例不需要的也可以不安装:

sudo apt-get install libcudnn8=${cudnn_version}-1+${cuda_version}sudo apt-get install libcudnn8-dev=${cudnn_version}-1+${cuda_version}sudo apt-get install libcudnn8-samples=${cudnn_version}-1+${cuda_version}

cuda版本是11.6 没有在最新的包管里器有的,我们用下载安装包deb的方式安装 。

下载好后安装,则会提示

sudo dpkg -i /mnt/f/downloads/cudnn-local-repo-ubuntu2204-8.5.0.96_1.0-1_amd64.deb
sudo cp /var/cudnn-local-repo-ubuntu2204-8.5.0.96/cudnn-local-7ED72349-keyring.gpg /usr/share/keyrings/
cd /usr/share/keyrings/sudo mv cudnn-local-7ED72349-keyring.gpg cuda-archive-keyring.gpgsudo apt updatesudo dpkg -i /mnt/f/downloads/cudnn-local-repo-ubuntu2204-8.5.0.96_1.0-1_amd64.deb

现在还没有完成,进入安装后的目录。

cd /var/cudnn-local-repo-ubuntu2204-8.5.0.96/ && ls

sudo dpkg -i libcudnn8_8.5.0.96-1+cuda11.7_amd64.debsudo dpkg -i libcudnn8-dev_8.5.0.96-1+cuda11.7_amd64.debsudo dpkg -i libcudnn8-samples_8.5.0.96-1+cuda11.7_amd64.deb

安装libcudnn8-dev与libcudnn8-samples包后,如果乐意的话,还可以尝试编译一下cuDNN的示例代码。如果能成功编译并运行,代表你的CUDA工具链和cuDNN库全部正常工作

安装完成后,进行一下测试

cp -r /usr/src/cudnn_samples_v8/ ~ && cd cudnn_samples_v8/mnistCUDNN/make clean && make

出现这样的的错误

test.c:1:10: fatal error: FreeImage.h: No such file or directory1 | #include "FreeImage.h"| ^~~~~~~~~~~~~compilation terminated.>>> WARNING - FreeImage is not set up correctly. Please ensure FreeImage is set up correctly. <<<

则需要安装 libfreeimage 库

sudo apt-get install libfreeimage3 libfreeimage-dev
make clean && make

出现passed通过就安装完成了

下次写一下,WSL2 Ubuntu安装PyTorch机器学习框架

标签: #ubuntu的库