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及进行环境配置手把手教学
- 0,操作代码汇总
- 1,查看/更新nvidia驱动版本号
- 2,创建虚拟环境
- 3,查看cuda(cudatoolkit)和cudnn版本
- 4,安装cuda(cudatoolkit)和cudnn
- 5,安装pytorch,torchvision,torchaudio
- 6,验证是否安装成功
若需要使用不同版本的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 Server 2019能不能安装win10的UMP应用呢?如果可以怎么装?求大佬解答