深度学习第一步——Pytorch-Gpu环境配置:Win11/Win10+Cuda10.2+cuDNN8.5.0+Pytorch1.8.0(步步巨细,少走十年弯路)

news/2025/3/14 16:39:45/

博主已有:Pycharm+Anaconda

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

import torchprint(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深度学习环境的一个总结!大部分人想学编程都被配置环境劝退,这个过程 难免会遇到各种各样的坑,但也正是这些坑让我们不断成长!希望这篇博客可以帮助到你,早日进军深度学习!


http://www.ppmy.cn/news/30265.html

相关文章

K8S篇-安装nfs插件

前言 有关k8s的搭建可以参考:http://t.csdn.cn/H84Zu 有关过程中使用到的nfs相关的nas,可以参考: http://t.csdn.cn/ACfoT http://t.csdn.cn/tPotK http://t.csdn.cn/JIn27 安装nfs存储插件 NFS-Subdir-External-Provisioner是一个自动配置…

生产环境线程问题排查

线程状态的解读RUNNABLE线程处于运行状态,不一定消耗CPU。例如,线程从网络读取数据,大多数时间是挂起的,只有数据到达时才会重新唤起进入执行状态。只有Java代码显式调用sleep或wait方法时,虚拟机才可以精准获取到线程…

数字图像学笔记 —— 17. 图像退化与复原(自适应滤波之「最小二乘方滤波」)

文章目录维纳滤波的缺点约束最小二乘方滤波给一个实际例子吧维纳滤波的缺点 维纳滤波(Wiener Filter),虽然是一种非常强大的退化图像还原算法,但是从实验过程我们也发现它存在着致命的缺陷,那就是要求输入退化系统的 …

KNN学习报告

原理 KNN算法就是在其表征空间中,求K个最邻近的点。根据已知的这几个点对其进行分类。如果其特征参数只有一个,那么就是一维空间。如果其特征参数只有两个,那么就是二维空间。如果其特征参数只有三个,那么就是三维空间。如果其特征…

进程控制~

进程控制 (创建、终止,等待,程序替换) 进程创建: pid_t fork();父子进程,数据独有,代码共享,各有各的地址 pit_t vfork();父进程阻塞,直到子进程exit退出或者程序替换之…

shell基础(5)算数计算:运算语法、自增自减

文章目录1. shell算数运算的特点2. 运算符一览3. 运算语法3.1 整形运算3.2. 小数运算 ing4. 自增自减4.1. a与a4.2. 自加1. shell算数运算的特点 Shell 和其它编程语言不同,Shell 不能直接进行算数运算,必须使用数学计算命令。Shell只支持整数运算&#…

PDF 解析格式化输出 API 数据接口

PDF 解析格式化输出 API 数据接口 支持输出 TEXT HTML XML TAG,多种格式输出,超精准识别率。 1. 产品功能 通用的识别接口, 支持标准 PDF 文件解析;多种格式输出,支持 TEXT HTML XML TAG;HTML 包含完美排…

Pytorch处理数据与训练网络问题汇总(协同训练)

基础语法 模型训练 【Swin-Unet】官方代码预训练权重加载函数load_from() 实际上由于SwinUnet是一个encoder-decoder对称的结构,因此加载权重时,作者并没有像通常那样仅仅加载encoder部分而不加载decoder部分,而是同时将encoder的权重对称地…