安装pytorch（附带踩坑过程）

首先已经安装过Anaconda(此处带过）

博主配置 amd的cpu+英伟达的2060显卡，python版本3.11.0

先把踩坑过程简述一遍

博主有两个环境一个是3.11.0一个是3.9.8

第一次，本机使用的是3.9.8，但是在Anaconda中create的是3.9.0。顺利下载完毕，import torch报错没有这个包。查询了大家的经验，可能是版本对不上使得没下载到这个包。

第二次，create的是3.9.8这个版本，报错没有查询到这个相关的版本

第三次，使用3.11.0的版本，下载并能使用-----成功

1、查看自己的环境下是什么版本的python

打开cmd控制台

python --version

2、创建环境

打开Anaconda Prompt

查看当前所有环境

创建一个新命名的环境，后缀带刚刚在cmd查到的版本

conda info --envs
创建环境和所需的python版本
conda create -n pytorch python=3.11.0
激活环境/切换环境
conda activate pytorch
查看环境下的包
pip list

3、查看所对应的cuda版本

先查询自己的GPU型号和支持的版本

驱动版本

打开cmd输入nvidia-smi

确认驱动版本大于398.26（小于需要升级，可以找驱动升级软件，这里就带过） version对应着支持cuda的版本，这里博主版本12.2，装的是11.3也可以用

4、下载pytorch

PyTorch

Package：如果是linux选择pip，如果是windows选择conda

conda install pytorch torchvision torchaudio pytorch-cuda=11.8 -c pytorch -c nvidia

成功后会有done的提示

再去pip list查看下是否安装成功了

确认有这个库尝试验证

python
import torch
torch.cuda.is_available()

显示True则安装成功。