1.pytorch官网查看cuda版本对应的torch和torchvisionde 版本(ncvv -V,nvidia-sim查看cuda对应的版本)

2.离线下载对应版本,网址https://download.pytorch.org/whl/torch_stable.html

我下载的:

cu113/torch-1.12.0%2Bcu113-cp37-cp37m-win_amd64.whl
cu113/torchvision-0.13.0%2Bcu113-cp37-cp37m-win_amd64.whl

cu是cuda,即GPU版本;1.12是torch版本/0.13是torchvision版本;cp37是python 3.7, win系统, 64位

3.虚拟环境 cd进入whl文件夹,pip  install 安装

4.验证环境;

#终端下验证
>>>import torch
>>>torch.cuda.is_available()
# True

坑和tips:

conda install 在线安装下来的都是CPU版本,无一例外

参考:

CUDA版本不对重装+离线快速下载torch的gpu版本安装+各种坑_Bellwen的博客-CSDN博客_重装cuda

https://download.pytorch.org/whl/torch_stable.html

Previous PyTorch Versions | PyTorch

离线安装Pytorch 最简单 高效的方法_正在学习的浅语的博客-CSDN博客_pytorch离线安装

pytorch最快的离线安装教程~10分钟完成,完不成我在线帮你~~_pytorch离线安装_heyque的博客-CSDN博客

Pytorch cpu版安装及卸载详细教程(以及安装成功后无法在ide中使用问题解决方法)_JDSYDWR的博客-CSDN博客_torch怎么卸载

更多推荐