(2020年)用官网condn命令安装时pytorch始终下载不下来。
我用的版本:cuda9.0 python3.6 pytorch==1.0.1 torchvision==0.2.2
由于先安装了cuda9.0,只好安装老版本,最新可以直接在官网对照版本安装
更新源:conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/win-64/
直接指定版本下载:conda install pytorch==1.0.1 torchvision==0.2.2
测试安装完成:
import torch
import torchvision
print(torch.cuda.is_available())
返回true,如果是返回false则是CUDA版本不对