Win10使用conda安装pytorch,torch.cuda.is_available()报错
在看了很多教程后使用conda安装pytorch,使用的cuda是10.2版本,查看pytorch版本后发现下载版本错误,根本原因是因为使用了国内镜像下载
输入如下
conda install pytorch torchvision cudatoolkit=10.2 -c pytorch
根据教程删去-c pytorch
使用镜像进行下载,但是镜像下载完之后为-cpu版本
需要使用官方链接进行下载
更改如下
另2020.7.11并无11.0版本对应的pytorch,下载安装cuda需要10.2版本