cuda 和torch的对应关系
时间: 2025-03-04 18:20:34 浏览: 67
### 查找 CUDA 和 PyTorch 版本之间的兼容性
为了确保所使用的 PyTorch 版本能够正常工作于特定的 CUDA 环境下,可以参照如下表格来匹配合适的版本组合。此表基于官方文档中的建议配置[^1]。
| CUDA Version | Compatible PyTorch Versions |
|--------------|-----------------------------|
| 9.0 | 1.0.0, 1.0.1 |
| 9.2 | 1.0.0, 1.0.1, 1.1.0 |
| 10.0 | 1.1.0, 1.2.0, 1.3.0, 1.3.1 |
| 10.1 | 1.3.0, 1.3.1, 1.4.0, 1.5.0 |
| 10.2 | 1.7.0, 1.8.0, 1.8.1, 1.9.0 |
| 11.0 | 1.7.0, 1.8.0, 1.8.1, 1.9.0 |
| 11.1 | 1.9.0 |
| 11.3 | 1.10.0 |
当安装 PyTorch 时,推荐使用官方网站提供的安装命令,并选择与当前系统中已安装的 CUDA 驱动相适应的选项[^2]。例如:
对于拥有 CUDA 11.3 的环境来说,可以通过 pip 安装最新稳定版 PyTorch 如下所示:
```bash
pip install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu113
```
阅读全文
相关推荐





