Torch1.7.1都和哪些版本的cuda,cudnn匹配多久几个例子
时间: 2024-04-03 18:37:19 浏览: 75
与cuda8.0匹配的cudnn5.1 方便大家下载
以下是 PyTorch 1.7.1 和 CUDA、cuDNN 的匹配版本列表:
| PyTorch 版本 | CUDA 版本 | cuDNN 版本 | NVIDIA 驱动程序版本 |
|--------------|-----------|------------|---------------------|
| 1.7.1 | 10.2 | 7.6 | 440.33 |
| 1.7.1 | 10.1 | 7.6 | 450.80.02 |
| 1.7.1 | 10.1 | 7.6 | 450.36.06 |
| 1.7.1 | 10.0 | 7.6 | 410.48 |
注意,这些版本号只是指 PyTorch 官方建议的匹配版本。实际上,可能还有其他版本的 CUDA、cuDNN 和 NVIDIA 驱动程序可以与 PyTorch 1.7.1 兼容,但需要进行一些额外的配置和测试工作。如果你需要使用特定版本的 CUDA、cuDNN 或 NVIDIA 驱动程序,请务必仔细阅读官方文档,并进行相应的测试和验证。同时,也建议你尽可能使用最新版本的软件和驱动程序,以获得更好的性能和稳定性。
阅读全文