win10+python3.8+pytorch安装(避坑)
安装gpu版本的pytorch
1.首先查看自己电脑是否支持
桌面右键–NVIDIA控制面板–帮助–系统信息–组件
这里我电脑最高支持的cuda版本是11.0.2
2、安装cuda&cudnn(也有大佬说不用安装,在安装pytorch时会自动安装)
直接下载安装就行
安装对应的cudnn
注意配置环境
这里推荐https://www.cnblogs.com/guan-zl/p/12986253.html的安装教程(pytorch安装没有采用该教程)
3、直接去pytorch官网下载相应版本
https://pytorch.org/get-started/previous-versions/
下载安装时,最好以管理员身份运行cmd,参考https://blog.csdn.net/iiiliuyang/article/details/123106024
**在官网提供的安装命令pip 和install之间加上–no-cache-dir install ,在命令结尾加上–user,**下载过程有点慢。
(这一步主要是为了防止在下载安装pytorch中电脑出现死机的状况)
Tip:
网上很多教程是用清华源镜像安装的,但是我在安装过程中发现torch.cuda.is_available()命令结果始终为FALSE,查看安装的包发现,清华源镜像给我安装的是cpu0版本的pytorch。