废话不多说,直接开始吧
1.进入官网
点击以往版本发现只有v1.2有cuda10选项
2. 打开Anaconda Prompt
新建pytorch环境
(base) C:\Users\RUAN>conda create -n pytorch python=3.6
建好之后启动
(base) C:\Users\RUAN>conda activate pytorch
安装pytorch
# CUDA 10.0
conda install pytorch==1.2.0 torchvision==0.4.0 cudatoolkit=10.0 -c pytorch
3.验证是否装好GPU
print torch.cuda.is_available()
输出
True
成功了~