资讯

展开

配置深度学习环境:在Windows(Win10)中安装CUDA,CUDNN,Pytorch-GPU版

作者:快盘下载 人气:

安装需知;NVIDIA GPU Driver;与硬件交互的低层软件;决定CUDA版本;CUDA版本决定cuDNN版本

①查看适配Driver的Cuda版本

在这里默认已安装Nvidia GPU Driver;一般电脑中只要有GPU就已安装了Driver
桌面 --》 鼠标右键 --》NVIDIA控制面板
配置深度学习环境:在Windows(Win10)中安装CUDA,CUDNN,Pytorch-GPU版
NVIDIA控制面板–》系统信息

组件 --》NVCUDA64.DLL 在产品名称栏可以看到CUDA 11.5.76 driver
则我的电脑适配的是Cuda 11.5及以下的版本
GPU版

②卸载已安装的Cuda

设置 --》应用 --》应用和功能
GPU版
在搜素中查找 NVIDIA,注意在显示列表中仅卸载含CUDA的软件即可;仅卸载含CUDA的软件即可;仅卸载含CUDA的软件即可;请慎重卸载
CUDA,CUDNN,Pytorch

③下载对应Cuda

进入Cuda工具包下载页面;选择适合自己版本的Cuda,我在这里选择11.5.0的版本
GPU版
选择系统类型
CUDA,CUDNN,Pytorch
选择软件包类别;点击download
CUDA,CUDNN,Pytorch

④安装Cuda

Cuda默认会安装到C盘/Program Files/NVIDIA Corporation中
因C盘容量不足;我准备将Cuda安装在D盘 新建了以下目录文件
CUDA,CUDNN,Pytorch
目录结构如下图所示;四个文件夹均为手动新建;此后安装时将路径选在这边
CUDA,CUDNN,Pytorch
双击打开我们之前下载的Cuda 11.5.0的安装包
CUDA,CUDNN,Pytorch
弹出如下界面;表示将生成一个临时文件;路径可以修改在其他位置;点击OK
GPU版
等待进度达到100%
CUDA,CUDNN,Pytorch
之后将弹出如下界面
CUDA,CUDNN,Pytorch
GPU版
点击同意并继续
CUDA,CUDNN,Pytorch
由于有些东西是不需要的;因此选择自定义安装
CUDA,CUDNN,Pytorch
勾选如下文两张图所示的内容
GPU版
GPU版
点击下一步;选择安装位置;下面是默认的安装位置;我要安装在D盘因此点击浏览;选择到我们之前新建的文件位置;注意文件名称与官方中的保持一致·
GPU版
为保持一致;在CUDA的目录中新建一个v11.5的文件夹
GPU版
其他两个路径同理;修改后如下图所示
GPU版
打钩;点击Next
CUDA,CUDNN,Pytorch
CUDA,CUDNN,Pytorch
GPU版
完成后如下图所示;点击关闭即可
GPU版
还记得之前生成的临时文件吗;这个时候就已经自动删除了

⑤下载Cudnn

Nvidia Cudnn下载网址;需登录nvidia账号才能下载;
由于我的Cuda版本为11.5 因此下载 根据自身需要下载xxxx for CUDA 11.5的Cudnn
CUDA,CUDNN,Pytorch
选择zip版进行下载
CUDA,CUDNN,Pytorch
下载完成后进行解压
CUDA,CUDNN,Pytorch

⑥安装Cudnn

下图中内容为Cudnn压缩包解压出的内容;全选复制
GPU版
粘贴在Cuda安装时的目录 NVIDIA GPU Computing ToolkitCUDAv11.5中
CUDA,CUDNN,Pytorch

⑦验证Cuda是否安装成功

使用如下命令;出现版本信息即代表安装成功

nvcc -V

CUDA,CUDNN,Pytorch

⑧安装Pytorch GPU版

在安装之前建议将源换为清华源或其他源;加快下载过程

进入Pytorch官网 下载Pytorch
CUDA,CUDNN,Pytorch
带有CUDA字样的就是GPU版;由于我们已安装CUDA的版本是11.5;因此我们可以下载小于等于11.5的版本
在anaconda中先激活需要安装pytorch的环境;我这里是之前创建过的名为d2l的环境
GPU版
在这里可以查看Pytorch的历史版本
CUDA,CUDNN,Pytorch
在这里我要安装1.11.0的Pytorch;选择CUDA11.3的版本
GPU版
复制该条指令 到anaconda中;回车执行
CUDA,CUDNN,Pytorch
确认安装 输入y
GPU版
等待下载完成
GPU版
出现以下界面代表安装成功
CUDA,CUDNN,Pytorch
使用conda list 查看已安装的包
GPU版
发现pytorch torchVision torchaudio cudatoolkit均在其中;至此;安装成功;完结撒花喽~~~

加载全部内容

相关教程
猜你喜欢
用户评论
快盘暂不提供评论功能!