博主已有:Pycharm+Anaconda

通过这篇博客你将获得:Cuda10.2+cuDNN11.x+Pytorch1.8.0(GPU)

import torch
 
print(torch.cuda.is_available())
print(torch.__version__)
print(torch.version.cuda)
True
1.8.0
10.2

目录

1.确定自己电脑有无显卡

2.确定显卡支持的cuda最高版本

​3.下载安装cuda

4.安装cuDNN 

5.安装Pytorch(重要!易踩坑!)

6.pycharm切换不了环境问题

7.总结


    今天是三月1日,开学的第一周,这个学期准备进军深度学习,主打的框架就是PaddlePaddle和Pytorch,PaddlePaddle都好说,有线上的Aistudio,头疼的是Pytorch,作为深度学习最经典的框架,肯定是必学无疑的。学习Pytorch,第一步就是配置环境,gpu和cpu的pytorch代码完全是两个东西,我之后想的是代码写好用云服务跑,所有代码得用gpu版本的,所以本地的Pytorch必须得是Gpu写的。实话实话,这中间我遇到了很多坑,所以我才想写这篇博客,帮大家以最快速的方式配制好深度学习的环境,话不多说,请看教程:

1.确定自己电脑有无显卡

搜索框搜索设备管理器进入,向下查找找到显示适配器: 

nvidia开头的即为电脑的显卡,我这里是MX450。

2.确定显卡支持的cuda最高版本

打开windows搜索框,搜索nvidia control panel,点击nvidia control panel进入:

点击系统信息:

点击组件,查看自己显卡支持Cuda最高版本,我这里最高支持11.1.96,所以我安装11.1.96以下版本的cuda都可以,在本篇博客中,安装的是10.2版本。

3.下载安装cuda

进入cuda官网:cuda官网

下载10.2版本的cuda,点击进入:

依次选择Windows,x86_64,10(win11系统可以兼容win10版本的cuda),local版本,网络版经常安装不成功。报错:nvidia安装程序无法继续未成功完成下载。

 Download下载base installer,下面两个是补丁,可以下载也可以不下载,本篇博客没有下载。 

  • 装补丁1(发布于2020年8月26日,选装)此修补程序解决了CUDA 10.2中捆绑的cuBLAS库中的一个问题,该问题在罕见的极端情况下导致数据无提示损坏。
  • 补丁2(发布于2020年11月17日,选装)此修补程序解决了cuBLAS库批处理的GEMM APIs中的一个问题,该问题在混合精度和快速数学的大批量计数的罕见情况下,导致数据无提示损坏。

之后按照指示安装就可以了,在安装选项时选择自定义安装:

 我没有改选项,直接下一步安装成功,如果出现问题可以看这篇博客:win10/win11+NVIDIA MX350笔记本自带显卡+CUDA10.2+Python3.6+Anaconda3从零开始配置深度学习环境

这里注意要将安装位置记录下来,之后安装cuDNN时候会用到:

 等待一段时间,之后检验一下是否安装完成,win+r,输入cmd,在窗口中输入nvcc -V

出现上述信息即为安装成功,注意nvcc -V的V需要大写,中间有个空格。

4.安装cuDNN 

在英伟达的官网下载与cuda版本相应的cuDNN版本,官网链接为:cuDNN

 点击download cuDNN,中间会让你填写个人信息,按照表格规范填写就可以了。

 点击历史版本

 下载匹配cuda10.2的,也就是8.5.0这个版本:

下载下来时是一个压缩包,解压,将cudnn文件夹里面的bin、include、lib文件直接复制到CUDA的安装目录下。这里的安装目录就是之前自定义cuda安装的位置:

5.安装Pytorch(重要!易踩坑!)

我大部分时间都花在安装这上面了,看了很多教程,不管是conda安装还是pip安装,最终的torch都是cpu版本的!

当时真的是百思不得其解,以为是自己的操作问题,不停的创建新环境,无论重装多少次,都一样,torch.cuda.is_available()返回false!终于在在一篇博客中找到了答案:torch.cuda.is_available()返回false——解决办法

因为我被conda镜像安装给坑了。我以为下载的是GPU版本,其实镜像下载的是cpu版本,所以我必须手动去官网用pip下载安装。

接下来请按照我的步骤安装,一步都不能错!

首先进入pytorch官网:Pytorch官网,点击Get Started,点击install previous versions of Pytorch,查看历史版本

 找到10.2cuda对应的torch,torchvision,torchaudio版本,记住这个版本号

pip install torch==1.12.1 torchvision==0.13.1 torchaudio==0.12.1

需要用pip下载,但是如果直接用pip指令在官网下载,速度非常慢导致torch下载失败,所有我们需要先下载.whl文件,再在本地安装!

.whl文件下载地址:torch_stable

cpu开头的是cpu版本的,可以直接跳过,我们看 cu开头的,cu102表示cuda10.2,cp38对应python3.8,按照需求下载就可以了。

如果没有对应的适配windows的文件包,这时候就可以去下载和自己cuda版本适配的较低版本的torch的搭配,记住版本号然后再去下载!

先用conda创建虚拟环境,win + R cmd进入控制台

conda create -n Pytorch-Gpu python=3.8
conda activate Pytorch-Gpu 

在刚才的虚拟环境中安装之前下载好的.whl文件,先安装torch的.whl文件,再安装另外两个。

至此,Pytorch的环境就算安装完成了,可以在控制台进行测试!

win+R cmd 进入控制台,先激活虚拟环境:

conda activate Pytorch-Gpu

 进入Python,导入torch库,查看cuda是否能用:

import torch
print(torch.cuda.is_available())
print(torch.__version__)
print(torch.version.cuda)

 True了,说明已经成功安装了!至此Pytorch就算完成安装了。

6.pycharm切换不了环境问题

我本以为完成上一步就算是打工告成了,现在虚拟环境已经能用了,只要用Pycharm选择这个虚拟环境就可以了,但我万万没想到,这个虚拟环境怎么也切换不过去!如果你出现了我上述这种情况,那么可以借鉴我下面这种做法。

之前我们在cmd控制台能够正常运行,那么在pycharm的终端也应该可以,所有在Pycharm首先激活

再运行一下,True了,说明没问题了! 出现这个问题可能是因为我在安装的时候选择的cuda版本超过了支持的cuda版本,后面我也会重新创建一个环境再走一遍流程的!

7.总结

这篇博客可谓是良心制作!一步一步,自己感觉很详细,质量也很高,算是对配置Pytorch深度学习环境的一个总结!大部分人想学编程都被配置环境劝退,这个过程 难免会遇到各种各样的坑,但也正是这些坑让我们不断成长!希望这篇博客可以帮助到你,早日进军深度学习!