直接按官网命令下载torch文件太慢,有时候还可能下一半直接中断导致下载失败。。
我们到https://download.pytorch.org/whl/torch_stable.html这个网站里:
cu+序号后面表示cuda版本,即GPU版本(cpu+序号表示cpu),如cu117表示cuda 11.7;
cp+序号表示python版本,如cp310表示python 3.10;
我们按自己的cuda版本和python版本找到对应的torch和torchversion文件即可,下载贼快。找到对应文件下载好放入需要配置torch的文件夹下:
然后是pip install + 文件名(注意要带whl),如:
pip install torchvision-0.14.0+cu117-cp310-cp310-win_amd64.whl
安装好后,来看安装效果:
获得一个令人安心的True。。
然后我们就可以愉快得使用torch环境啦!
文章出处登录后可见!
已经登录?立即刷新