Windows下的深度学习环境安装

发布时间 2023-03-22 21:27:22作者: Sataned

Windows下的深度学习环境安装

电脑型号:戴尔G15

GPU:RTX3060

第一步:下载anaconda

选择anaconda3 2022.10月版本最新版的windowsx86.exe版。官网下载特别慢,这里采用镜像网站

Index of /anaconda/archive/ | 北京外国语大学开源软件镜像站 | BFSU Open Source Mirror

​ 或者参考这篇文章:

Anaconda之conda换国内源_你的大数据学长的博客-CSDN博客_conda换源

​ 在进入页面之后:

image-20230322204227122

第二部:安装anaconda

选择好安装路径,之后直接安装

image-20230322204259629

第三步:创建pytorch环境

​ 输入conda create -n pytorch python=

​ pytorch是你指定的环境名称 python后面对应的是你需要的版本,之后输入y确定安装

输入pip3 install opencv-python自动安装最新版本的opencv和numpy

image-20230322204326474

​ 之后打开pytorch官网,根据自己的配置选择环境,我这里是选择了CUDA11.7

image-20230322204457247

第四步:\1. 装配cudaTookits

​ 打开网站CUDA GPUs - Compute Capability | NVIDIA Developer,我的是GeForce系列的RTX3060,看自己的显卡是否在里面

image-20230322204613833

​ 安装cudatookits:CUDA Toolkit Archive | NVIDIA Developer

image-20230322204630496

image-20230322204634410

​ 看这个应该是兼容11.6以上的,我安装的11.7的所以就直接安装了最新版本的了。安装页面如下:我选择了E盘安装,反正别在C盘安装就对了

image-20230322204646437

​ 等待安装成功

image-20230322204655872

image-20230322204700117

直接全部选择下一步安装即可,时间有点长,耐心等待。

安装完成后输入nvcc -V查看是否安装成功,下图为安装成功的页面

image-20230322204708807

第五步:安装cudnn

​ 网站https://developer.nvidia.com/cudnn

image-20230322204739788

​ 注册一个账号然后登陆

image-20230322204750468

看好你之前安装的cuda版本,我这里是11.7所以选择了第一个

image-20230322204758476

image-20230322204804564

​ 下载之后,解压缩,将CUDNN压缩包里面的bin、clude、lib文件直接复制到CUDA的安装目录下(C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1),直接覆盖安装即可完成。在电脑中添加环境变量:C:\Program Files\NVIDIA Corporation\NVSMI

image-20230322204822824

image-20230322204826020

image-20230322204832244

image-20230322204840947

image-20230322204849055

​ 之后输入如下几条代码,如图。第一个是版本,第二个是是否开启了GPU模式。如果结果和我类似则安装成功,若第二个为false则需要去看看你的显卡支持的cudatoolkit版本是否有问题。如上图我的Driver Version为527.41,然后进入cudatoolkit版本页面看里面支持这个版本号需要大于多少,选择你能满足的版本。之后也要注意你的pytorch版本是否和cuda版本对应和你的cudnn版本对应,牵一发而动全身。

image-20230322204901206

*总结*

这次安装就是一次过了,除了中间安装cudnn的时候c盘内存炸了一下重新分盘耗了点时间,差不多两个小时安装完毕,中间没有出现任何的bug,这次比第一次安装要顺利多了。如果你安装的时候依旧有问题的话,不要着急,第一次我安装了得有好几天,挺折磨人的,多去CSND或者b站找找遇到的类似的问题,有了一次经验以后之后再安装就很方便了。

参考网站:

  1. 面向Windows的Pytorch完整安装教程_钱彬 (Qian Bin)的博客-CSDN博客_pytorch windows安装

​ 2.Anaconda之conda换国内源_你的大数据学长的博客-CSDN博客_conda换源