2024pytorch小土堆安装教程（GPU），踩坑总结

目录

2024pytorch小土堆安装教程（GPU），踩坑总结

本人小白，应老师要求自学深度学习，跟着b站up小土堆老师学习深度学习，第一节课是安装pytorch,在历经三次失败后才获得成功，于是决心写下自己的总结以供大家参考借鉴。以下均基于小土堆的PyTorch深度学习快速入门教程的内容上得到，希望能对有同样问题的朋友们有所帮助。

一、安装Anaconda

这一步较简单，一般不会出问题。我在此之前学习数据预处理就已经安装好了，小土堆的视频有详细讲述如何下载Anaconda，此处不再赘述。
另外由于之前下载的Anaconda版本过低，在安装pytorch的时候提醒我安装，亲测不用更新版本也可以正确安装。

二、检查显卡及驱动安装

根据提供的三种方法查看电脑的显卡即可，若nvidia开头则后续pytorch安装中选择GPU版本即可，如无选择CPU版本。

三、创建虚拟环境

输入以下代码创建虚拟环境,-n代表名称
conda create -n 虚拟环境名称 python=版本
踩坑一：创建对应的python版本。小土堆的python版本为3.6，所以视频中他的代码为

conda create -n pytorch python=3.6 

但每个人的版本不同，可以win+R打开命令行，输入python查看版本

可以看到我的版本即为3.9
后续输入y继续下载即可，下载完成后输入conda activate 环境名称进入我们新的环境（ps：删除某个环境一定不能在该环境下，需要切换到别的环境）

四、安装pytorch

同视频教程，需要去官网找到你所需要版本的对应指令，官网链接：pytorch
在此之前需要查看CUDA driver 版本：win+R打开命令行，输入nvidia-smi查看版本

踩坑二：需要保证：

1. Driver Version 大于396.26，如果小于需要更新，可查看视频进行更新
2. CUDA Driver 版本(如上图，我的是11.6)>=CUDA Runtime 版本(即cudatoolkit后面跟的版本数)，如果首页下滑没有所需要的版本，可以进入历史版本寻找。我下载的版本如下：
   踩坑三：网速
   由于官网是国外网站，所以下载速度很慢，而且容易出错，出错情况例如：
   这是由于下载包太长导致的，有三种解决方法，此处推荐镜像下载。即在我们刚刚得到的指令中删去指令尾部的pytorch,加上镜像源网址即可，此处我用的是清华源，例如：

conda install pytorch==1.12.1 torchvision==0.13.1 torchaudio==0.12.1 cudatoolkit=11.3 -c https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/win-64/

五、检查验证

同视频，输入

pip list

查看是否有torch，如图：

随后顺序输入

python

import torch

 torch.cuda.is_available()

最后若返回True则下载成功！