CUDA12对应的pytorch和cuDNN版本
时间: 2023-09-23 18:04:13 浏览: 5331
CUDA 12.4官方文档 - 英文
对于CUDA 12,可以使用以下版本的PyTorch和cuDNN:
- PyTorch 1.9.0 或更高版本
- cuDNN 8.2.0 或更高版本
请注意,cuDNN版本也应该与CUDA版本相匹配。例如,如果您使用CUDA 12.0,则应该使用cuDNN 8.0.x版本。建议您查看PyTorch的官方网站,以获取有关特定版本的PyTorch和cuDNN的更多信息。
阅读全文