成功下载gpu版本pytorch
作者:互联网
在下载pytorch的时候遇到很多问题:
1、从官网下载速度太慢,容易下载中断
2、使用清华源下载很快,但是我下载到的都是cpu版本的,哪怕我明明下载的是gpu版本
3、从官网下载轮子文件,因网速太慢容易中断;从终端下载也会因为网速问题而中断
最后我用下面的命令成功下载了gpu版本的pytorch!
下载的时候自动使用清华源,torch版本和torchvision版本自定义。
pip install torch==1.7.0 torchvision==0.8.0 -f https://download.pytorch.org/whl/torch_stable.html -i https://pypi.tuna.tsinghua.edu.cn/simple
标签:torch,网速,pytorch,版本,gpu,下载 来源: https://blog.csdn.net/weixin_48158964/article/details/120710259