PyTorch 1.7.1 的CUDA10.1和cuDNN7.6版本需要匹配哪个NVIDIA的驱动程序
时间: 2024-04-03 21:37:17 浏览: 135
ubuntu cuda_10.1 cudnn7.6安装包 显卡驱动
PyTorch 1.7.1 的CUDA 10.1 和 cuDNN 7.6 版本需要匹配的NVIDIA驱动程序版本是 450.x。具体来说,你需要安装一下版本号的驱动程序:
- 对于Linux系统:NVIDIA驱动程序版本为 450.xx。
- 对于Windows系统:NVIDIA驱动程序版本为 456.38。
需要注意的是,这个版本的驱动程序只适用于支持CUDA 10.1版本的NVIDIA GPU。如果你的GPU不支持CUDA 10.1版本,则需要安装适合你的GPU的驱动程序。可以在NVIDIA官网上查找匹配的驱动程序。
阅读全文