win10+python3.8+pytorch安装(避坑)

win10+python3.8+pytorch安装(避坑)

安装gpu版本的pytorch

1.首先查看自己电脑是否支持

桌面右键–NVIDIA控制面板–帮助–系统信息–组件
win10+python3.8+pytorch安装(避坑)_第1张图片
这里我电脑最高支持的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/
win10+python3.8+pytorch安装(避坑)_第2张图片

下载安装时,最好以管理员身份运行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。

你可能感兴趣的:(pytorch)