TensorFlow、PyTorch各版本对应的CUDA、cuDNN关系(截止2021年4月7日)
- TensorFow
-
- PyTorch
-
- cuDNN与CUDA
- CUDA和NVIDIA显卡驱动关系
因为之前tensorflow、pytorch配环境在cuda上吃过很多亏,主要是版本不匹配造成的,所以自行整理了对应关系,资料均来自官网。
TensorFow
官网:https://www.tensorflow.org/?hl=zh-cn
Linux
CPU
GPU
macOS
CPU
GPU
Windows
CPU
GPU
PyTorch
PyTorch官网:https://pytorch.org/
PyTorch历史版本:https://pytorch.org/get-started/previous-versions/
CPU
PyTorch | torchvision |
---|
1.8.0 | 0.9.0 |
1.7.1 | 0.8.2 |
1.7.0 | 0.8.0 |
1.6.0 | 0.7.0 |
1.5.1 | 0.6.1 |
1.5.0 | 0.6.0 |
1.4.0 | 0.5.0 |
1.2.0 | 0.4.0 |
1.1.0 | 0.3.0 |
1.0.1 | 0.2.2 |
1.0.0 | 0.2.1 |
GPU
PyTorch | CUDA | torchvision |
---|
1.8.0 | 10.2、11.1 | 0.9.0 |
1.7.1 | 9.2、10.1、10.2、11.0 | 0.8.2 |
1.7.0 | 9.2、10.1、10.2、11.0 | 0.8.0 |
1.6.0 | 9.2、10.1、10.2 | 0.7.0 |
1.5.1 | 9.2、10.1、10.2 | 0.6.1 |
1.5.0 | 9.2、10.1、10.2 | 0.6.0 |
1.4.0 | 9.2、10.1 | 0.5.0 |
1.2.0 | 9.2、10.0 | 0.4.0 |
1.1.0 | 9.0、10.0 | 0.3.0 |
1.0.1 | 9.0、10.0 | 0.2.2 |
1.0.0 | 8.0、9.0、10.0 | 0.2.1 |
cuDNN与CUDA
官网:https://developer.nvidia.com/rdp/cudnn-archive
cuDNN | CUDA |
---|
8.1.0 | 11.2、11.1、11.0、10.2 |
8.0.5 | 11.1、11.0、10.2、10.1 |
8.0.4 | 11.1、11.0、10.2、10.1 |
8.0.3 | 11.0、10.2、10.1 |
8.0.2 | 11.0、10.2、10.1 |
8.0.1 | 11.0、10.2 |
7.6.5 | 10.2、10.1、10.0、9.2、9.0 |
7.6.4 | 10.1、10.0、9.2、9.0 |
7.6.3 | 10.1、10.0、9.2、9.0 |
7.6.2 | 10.1、10.0、9.2、9.0 |
7.6.1 | 10.1、10.0、9.2、9.0 |
7.6.0 | 10.1、10.0、9.2、9.0 |
7.5.1 | 10.1、10.0、9.2、9.0 |
7.5.0 | 10.1、10.0、9.2、9.0 |
7.4.2 | 10.0、9.2、9.0 |
7.4.1 | 10.0、9.2、9.0 |
7.4.0 | 10.0、9.2、9.0 |
7.3.0 | 10.0、9.0 |
7.2.1 | 9.2 |
CUDA和NVIDIA显卡驱动关系
官网:https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html
本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系:hwhale#tublm.com(使用前将#替换为@)