废话不多说,直接开始吧
1.进入官网![在这里插入图片描述](https://img-blog.csdnimg.cn/20210701162936731.png?x-oss-process=image/watermark,type_ZmFuZ3poZW5naGVpdGk,shadow_10,text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzQzNjc4MDA1,size_16,color_FFFFFF,t_70)
点击以往版本发现只有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
成功了~