CUDA12对应的pytorch和cuDNN版本
时间: 2023-09-23 16:04:13 浏览: 3795
对于CUDA 12,可以使用以下版本的PyTorch和cuDNN:
- PyTorch 1.9.0 或更高版本
- cuDNN 8.2.0 或更高版本
请注意,cuDNN版本也应该与CUDA版本相匹配。例如,如果您使用CUDA 12.0,则应该使用cuDNN 8.0.x版本。建议您查看PyTorch的官方网站,以获取有关特定版本的PyTorch和cuDNN的更多信息。
相关问题
cuda12.2对应的cudnn和pytorch版本
对于CUDA 12.2,以下是推荐的cuDNN和PyTorch版本:
1. cuDNN:CUDA 12.2与cuDNN 8.3.0 兼容。您可以从NVIDIA官方网站下载相应版本的cuDNN。
2. PyTorch:对于CUDA 12.2,建议使用PyTorch 1.10及更高版本,因为这些版本已经对CUDA 12.2进行了适配。您可以通过PyTorch官方网站或者使用pip命令安装最新版本的PyTorch。
请注意,这些是推荐的版本,但不排除其他版本的兼容性。在安装和使用过程中,您还应该查看相关文档以确保正确的版本兼容性。
pytorch、cuda和cuDNN对应的版本
不同版本的PyTorch需要不同版本的CUDA和cuDNN。具体对应关系如下:
| PyTorch版本 | CUDA版本 | cuDNN版本 |
| ----------- | -------- | --------- |
| 1.0.x | CUDA 9.0 | cuDNN 7.3 |
| 1.1.x | CUDA 10.0 | cuDNN 7.4 |
| 1.2.x | CUDA 10.1 | cuDNN 7.6 |
| 1.3.x | CUDA 10.1 | cuDNN 7.6 |
| 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.0 |
| 1.8.x | CUDA 11.1 | cuDNN 8.0 |
| 1.9.x | CUDA 11.1 | cuDNN 8.2 |
需要注意的是,不同版本的CUDA和cuDNN也有对应的操作系统和GPU型号要求,具体可以参考PyTorch官方文档。