我现在是RTX 2080, CUDA VERSION 是10.1 ,请问GPU版的torch怎么安装。尽量使用国内的连接,因为国外很慢
时间: 2024-06-07 13:05:18 浏览: 213
你可以使用以下命令在命令行中安装GPU版的torch:
pip3 install torch torchvision -f https://download.pytorch.org/whl/cu101/torch_stable.html -i https://pypi.tuna.tsinghua.edu.cn/simple
这将从清华镜像站下载所需的文件,以加快下载速度。
阅读全文