【Pytorch】Pytorch（环境搭建+问题共享）

Pytorch 环境搭建

【写在前面】本文总体参考下文，同时针对个人的情况进行了额外的总结，请先阅读下文

文献标题： windows10安装pytorch，感觉自己基本踩了所有坑，这里提供最有效的两种方法
链接： https://blog.csdn.net/namespace_Pt/article/details/104235075.
提示： 若跳转失败，手动复制链接进行跳转

Pytorch 问题共享

情景一：电脑不支持CUDA

☸电脑不支持CUDA，且需要用到Pytorch的，根据上文目录中的3和5进行（见下图）

装好Anaconda3以后，在官网下载到对应版本的Pytorch，这里有以下几点需要注意

• 查看电脑是否支持CUDA，我下载了一个叫GPU-z的软件，CUDA前面不打勾说明不支持，百度有教程，但是会捆绑很多软件，很烦
• Anaconda3对应的是Python37，所以尽量直接下载Python37，避免后续踩一些不必要的坑，我在[强化学习平台“天授”环境搭建]中有提到，仅为个人观点
• 下载Python37对应的torch，torch如有版本要求，下载对应的版本，电脑不支持CUDA时，下载cpu开头的，如下图

由于我需要torch>=1.4.0版本的，所以对应如图

附上torch下载链接： https://download.pytorch.org/whl/torch_stable.html.

这里建议先把whl文件下载到本地，直接点击上面的链接即可，保存路径就放在上面创建的环境目录下，如我这里是 D:\anaconda3\envs\ pytorch \Lib\site-packages （pytorch就是刚才activate环境时自己命名的环境，创建好后就有这个文件夹了）

然后在环境pytorch中pip下载的torch，如图，注意结尾的文件类型whl不可丢

情景二：电脑支持CUDA，但找不到与CUDA版本匹配的torch

对于这种情况

torch下载链接：https://download.pytorch.org/whl/torch_stable.html.

如果在上述链接中都没有找到匹配的torch，那只能考虑给CUDA 降级（升级）了
此时，可参考我的另外一篇【NVCUDA】 降级（升级）