PyTorch环境配置(GPU版)-新手成功运行-2023年7月

前言：本人大二，纯小白，踩了很多坑，终于完整跑通小土堆的整个教程，遂写篇博客希望能帮到后人。

【得用Aconda先安装python】

本人配环境的最终情况【GPU版本】：

1.创建虚拟环境，python版本3.10

解释：pytorch目前好像只支持python3.8及以上。

https://pytorch.org/get-started/locally/#windows-python官方文档这里有说

接下来的操作均是在这一步建立的虚拟环境下执行的

2.更换镜像源为【阿里】的。进虚拟环境执行以下命令：

conda config --add channels https://mirrors.aliyun.com/pypi/simple/

解释：我之前是清华的镜像源，然后到后期用不了gpu来训练（网上查到好像是清华的pip只能下cpu版的？），把清华的都删了再下阿里的就能用...

3.到pytorch官网，用【pip命令】下载，对应自己显卡的CUDA版本。

解释：我先用conda命令下报错，后用pip命令顺利下载好。

4.试试import torch会不会报错，看看torch.cuda.is_available()返回结果是否为True

 解释：如果为False而且你是有显卡的，建议再搜别的办法吧、、不过也不影响运行也能用cpu来算，只是cpu可能没gpu算的快