torch cuda版本国内源安装

目前国内只有阿里云有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:

发表评论