首先,去Anaconda官网下载并安装Anaconda(这里如果C盘不是很满的话,建议装在C盘),这里推荐去它的中文网(也可以去它的官网) (Anaconda安装的时候提示打的√尽量都√上,选Just me)
安装完毕之后点开始,找到 Anaconda Prompt 并打开
然后使用以下命令在Anaconda下创建虚拟环境:
conda create -n pytorch-g python=3.6(-n后面的为自定义环境名,笔者可自行定义,这里我们安装的pytorch的gpu版本,所以命名为pytorch-g,python=3.6为python版本)
创建成功后使用一下命令激活环境:
conda activate pytorch-g
可以看见括号里的base环境变为pytorch-g
然后:
pip list
只有上图所示的几个包
接下来,安装Pytorch以及cuda(这里我推荐直接去Pytorch官网安装,因为最新的cuda11.6和11.7版本直接包含在Pytor