torch都和哪些版本的cuda,cudnn匹配
时间: 2024-04-04 20:29:01 浏览: 14
PyTorch和CUDA、cuDNN的版本匹配关系如下:
| PyTorch 版本 | CUDA 版本 | cuDNN 版本 |
|--------------|-----------|------------|
| 1.0.x | CUDA 9.0 | cuDNN 7 |
| 1.1.x | CUDA 10.0 | cuDNN 7 |
| 1.2.x | CUDA 10.1 | cuDNN 7 |
| 1.3.x | CUDA 10.1 | cuDNN 7 |
| 1.4.x | CUDA 10.1 | cuDNN 7.6 |
| 1.5.x | CUDA 10.2 | cuDNN 7.6 |
| 1.6.x | CUDA 10.2 | cuDNN 7.6 |
| 1.7.x | CUDA 11.0 | cuDNN 8 |
| 1.8.x | CUDA 11.1 | cuDNN 8 |
| 1.9.x | CUDA 11.1 | cuDNN 8 |
这个表格列出了常见的PyTorch版本与CUDA、cuDNN版本的匹配关系。需要注意的是,PyTorch 1.9版本只支持CUDA 11.1和cuDNN 8。如果你想使用其他版本的CUDA和cuDNN,需要安装对应版本的PyTorch。
此外,还需要注意的是,不同版本的CUDA和cuDNN需要匹配不同版本的NVIDIA驱动程序。你需要根据你的硬件和软件环境来选择正确的版本。你可以在NVIDIA官网上查找匹配的驱动程序。