1.下载轮子

进入网站下载对应cuda版本的torch、torchvision轮子

http://download.pytorch.org/whl/torch_stable.htmlicon-default.png?t=M7J4http://download.pytorch.org/whl/torch_stable.html

2.安装轮子

打开下载文件夹

Ctrl + Alt + T 打开终端

pip install torchvision-0.13.1+cu116-cp39-cp39-linux_x86_64.whl

pip install torch-1.12.1+cu116-cp39-cp39-linux_x86_64.whl

Logo

有“AI”的1024 = 2048,欢迎大家加入2048 AI社区

更多推荐