Torch1.7.1都和哪些版本的cuda,cudnn匹配
时间: 2024-04-04 14:28:58 浏览: 58
torch_cluster-1.5.8-cp37-cp37m-linux_x86_64whl.zip
PyTorch 1.7.1 的CUDA和cuDNN版本匹配关系如下:
| PyTorch 版本 | CUDA 版本 | cuDNN 版本 |
|--------------|-----------|------------|
| 1.7.1 | CUDA 10.1 | cuDNN 7.6 |
需要注意的是,这个表格列出了PyTorch 1.7.1版本与CUDA、cuDNN版本的匹配关系。如果你想使用其他版本的CUDA和cuDNN,需要安装对应版本的PyTorch。
此外,还需要注意的是,不同版本的CUDA和cuDNN需要匹配不同版本的NVIDIA驱动程序。你需要根据你的硬件和软件环境来选择正确的版本。你可以在NVIDIA官网上查找匹配的驱动程序。
阅读全文