gpu版torch安装

一般的教程

去PyTorch安装

一般的结果

只能装上cpu版的

原因

为什么conda安装pytorch版本不对-CSDN博客

解决方案

torch和torchvision 版本对应_torch 2.4 对应的torchvision-CSDN博客

比如我的电脑的安装命令是

conda install pytorch==2.4.0 torchvision==0.19.0 torchaudio==2.4.0 pytorch-cuda=12.4 -c pytorch -c nvidia

whl下载网站：

1. https://mirrors.aliyun.com/pytorch-wheels/cu117/

这个不同的cuda版本要改网址后缀

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

注意点

• 现在装pytorch不需要专门去装cudnn和toolkit了

记录 coda装pytorch并不需要单独装cuda和cudnn|李沐评论_torch需要提前装cuda吗-CSDN博客

• cuda向前兼容，如12.4的CUDA Version: 12.4可以装11.6的torch.version.cuda

查看已经安装的软件包列表

conda list
pip list