1.首先查看自己电脑的CUDA
win+r,输入cmd

命令行输入nvidia-smi,查看自己CUDA的版本,我的是10.2,所以在安装pytorch时,版本不能选择高于10.2的。

2.打开anaconda prompt,输入conda create -n pytorch python=3.6,python的版本你可以自己选择,我这里选择的是与anaconda中的python版本相同的

输入y之后会出现如下界面:

输入conda activate pytorch代码,进入pytorch环境
3.进入pytorch官网,我们一般都会选择10.2的版本,conda进行安装,但是命令显示10.2的版本不再支持windows系统,不要慌,我们可以通过下方进入以前的版本进行安装

输入以下命令进行安装,时间可能会有点长
conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch

按照提示走一下

显示done之后输入pip list看是否有torch相关的包

然后通过以下操作验证是否成功

本文介绍如何根据自己的CUDA版本选择合适的PyTorch版本进行安装。包括检查CUDA版本、创建虚拟环境、选择安装源以及验证安装是否成功的方法。
1191

被折叠的 条评论
为什么被折叠?



