PyTorch

official website

历史版本

这里可以找到指定版本的PyTorch（CUDA版本和CPU版本），在部署老项目时很有用：

历史版本

Q&A

• Q1: 在安装CUDA版本的PyTorch之前需要先去NVidia官网下载对应版本的CUDA吗？
  A1: pip安装方式下不需要，因为这种方式下CUDA版本的PyTorch包中已经集成CUDA了，其它安装方式（比如从源码编译安装）可能需要。

Info

关于Q1的实验：

首先查询本机之前安装的CUDA版本：

> nvcc --version

结果显示CUDA版本为12.9

接下来安装CUDA11.8版本的PyTorch：

> pip install torch==2.3.0 torchvision==0.18.0 torchaudio==2.3.0 --index-url https://download.pytorch.org/whl/cu118

结果可以安装成功，然后使用Python命令行进行交互：

>>> import torch
>>> torch.cuda.is_available()
True
>>> torch.version.cuda
'11.8'

结论符合A1的描述。 注意：此结论仅适用于pip安装方式，其它方式可能不适用（conda安装方式未测试，因此不确定）。