目前国内只有阿里云有cuda版本的torch,具体方法如下:
pip3 install torch==1.7.1 torchvision -f https://mirrors.aliyun.com/pytorch-wheels/cu110
注意torch版本需要指定,否则会自动选择最新的CPU版本,具体哪些可用可以直接打开-f后面的链接查看:
另外,判断的简单方法就是下载文件的大小,如果是cuda版本whl大小必然会大于1G,而CPU版本大概200M:
在虚构的故事中寻求真实感的人脑袋一定有问题
目前国内只有阿里云有cuda版本的torch,具体方法如下:
pip3 install torch==1.7.1 torchvision -f https://mirrors.aliyun.com/pytorch-wheels/cu110
注意torch版本需要指定,否则会自动选择最新的CPU版本,具体哪些可用可以直接打开-f后面的链接查看:
另外,判断的简单方法就是下载文件的大小,如果是cuda版本whl大小必然会大于1G,而CPU版本大概200M: