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怎么卸载