打开Anaconda Prompt
创建一个conda(也可以不用,这里为了管理方便)
conda create -n pytorch python=3.6
conda 是指调用 conda 包,create 是创建的意思,-n 是指后面的名字是屋子的名字, pytorch是屋子的名字(可以更改成自己喜欢的),python=3.6 是指创建的屋子,是 python3.6 版本。
之后,弹出提示,输入 y,即可安装。
安装成功后,输入以下指令查看环境:
conda info --envs
进入刚刚的环境安装pytorch
conda activate pytorch
进入官网
https://pytorch.org/
复制回车即可
验证是否成功(是否可以用GPU版本)
import torch
torch.__version__
torch.cuda.is_available()
如果为true,说明你可以用GPU版本