windows怎么装cudnn

Posted

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了windows怎么装cudnn相关的知识,希望对你有一定的参考价值。

安装cudnn前,得确保正确安装好cuda

首先到官网下载cudnn:Nvidia官网cudnn 选择下载版本时要注意和Cuda版本匹配(这里下载的是cudnn-for-cuda-8.0-windows7-x64-v5.1)。下载下来的是一个压缩包,解压后有3个文件夹:

将这些文件夹覆盖至CUDA的安装目录下

例如:C:\\Program Files\\NVIDIA GPU Computing Toolkit\\CUDA\\v8.0\\

即可完成安装。

参考技术A NVIDIA CuDNN 安装说明
CuDNN是专门针对Deep Learning框架设计的一套GPU计算加速方案,目前支持的DL库包括Caffe,ConvNet, Torch7等。
CuDNN可以在官网免费获得,注册帐号后即可下载。官网没有找到安装说明,下载得到的压缩包内也没有Readme. 不过google一下就会找到许多说明。基本原理是把lib文件加入到系统能找到的lib文件夹里, 把头文件加到系统能找到的include文件夹里就可以。这里把他们加到CUDA的文件夹下(参考这里)
tar -xzvf cudnn-6.5-linux-R1.tgz
cd cudnn-6.5-linux-R1
sudo cp lib* /usr/local/cuda/lib64/
sudo cp cudnn.h /usr/local/cuda/include/

执行后发现还是找不到库, 报错
error while loading shared libraries: libcudnn.so.6.5: cannot open shared object file: No such file or directory

而lib文件夹是在系统路径里的,用ls -al发现是文件权限的问题,因此用下述命令先删除软连接
cd /usr/local/cuda/lib64/
sudo rm -rf libcudnn.so libcudnn.so.6.5

然后修改文件权限,并创建新的软连接
sudo chmod u=rwx,g=rx,o=rx libcudnn.so.6.5.18
sudo ln -s libcudnn.so.6.5.18 libcudnn.so.6.5
sudo ln -s libcudnn.so.6.5 libcudnn.so本回答被提问者采纳

如何在anaconda虚拟环境中安装多个版本的CUDA,cudnn,pytorch,torchvision,torchaudio及进行环境配置手把手教学

如何在anaconda虚拟环境中安装多个版本的CUDA,cudnn,pytorch,torchvision,torchaudio及进行环境配置手把手教学

若需要使用不同版本的cuda,只需创建不同的虚拟环境,下载所需的cuda及cudnn版本即可。

0,操作代码汇总

以cuda11.3,cudnn8.2.1为例

查看/更新驱动上限

创建环境:
conda create -n cuda11_3_cudnn8_2_1_env python=3.8
激活环境:
conda activate cuda11_3_cudnn8_2_1_env
查询cuda版本:
conda search cudatoolkit --info
查询cudnn版本:
conda search cudnn --info
安装cuda:
conda install cudatoolkit=11.3
安装cudnn:
conda install cudnn=8.2.1
安装pytorch,torchvision,torchaudio:
conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch
验证:
conda list
python
import torch
torch.cuda.is_available()

返回true为成功

具体操作:

1,查看/更新nvidia驱动版本号

先更新驱动,后查看版本。也可以不更新,直接查看。

GeForce Experience可以更新为最新版本驱动

控制面板查看当前最高能支持的cuda版本
我这里最高支持11.7.99

2,创建虚拟环境

以管理员身份进入Anaconda Promot

conda create -n cuda11_3_cudnn8_2_1_env python=3.8


激活环境

conda activate cuda11_3_cudnn8_2_1_env

3,查看cuda(cudatoolkit)和cudnn版本

查看conda支持的cuda版本,选择合适的版本

conda search cudatoolkit --info


可以看到最新支持到11.3.1,要求cuda必须>=11.3,我是11.7,满足。

查看conda支持的cudnn版本,选择合适的版本

conda search cudnn --info


可以看到最新支持到8.2.1,要求cudatoolkit在11.0到11.4之间,我是11.3,满足。

4,安装cuda(cudatoolkit)和cudnn

conda install cudatoolkit=11.3

conda install cudnn=8.2.1

5,安装pytorch,torchvision,torchaudio

进入https://pytorch.org/get-started/locally/

选择对应版本,复制官方给的代码

conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch

6,验证是否安装成功

conda list

查看是否下载好对应版本
代码验证,进入python代码模式

python
import torch

没返回异常,说明pytorch装好了

torch.cuda.is_available()

返回true,说明cuda也装好了

以上是关于windows怎么装cudnn的主要内容,如果未能解决你的问题,请参考以下文章

有两个WINDOWS补丁总是装不上。

Windows Server 2019能不能安装win10的UMP应用呢?如果可以怎么装?求大佬解答

重装系统后怎么回到原来的系统?

在mac中安装windows由于u盘是fat32格式所以装不了4g的单独文件安装停止怎么

树莓派怎么装win10 lot

我装虚拟机提示要装IIS,IIS怎么装呢?