Tensorflow|PyTorch|Paddle|ONNXRuntime与CUDA、cuDNN对应汇总

Tensorflow版本对应关系

Linux GPU

PyTorch版本对应关系

PyTorch	CUDA
1.11.0	10.2、11.3
1.9.0	10.2、11.1、11.3
1.8.1	10.1、10.2、11.1、11.3
1.8.0	10.2、11.1
1.7.1	9.2、10.1、10.2、11.0
1.7.0	9.2、10.1、10.2、11.0
1.6.0	9.2、10.1、10.2
1.5.1	9.2、10.1、10.2
1.5.0	9.2、10.1、10.2
1.4.0	9.2、10.1
1.2.0	9.2、10.0
1.1.0	9.0、10.0
1.0.0	8.0、9.0、10.0

PaddlePaddle

PaddlePaddle	CUDA
2.2	10.1、10.2、11.0、11.1、11.2
2.0	9.0、10.0、10.1、10.2、11.0
1.8	9.0、9.1、9.2、10.0

ONNXRuntime

其他资料

• cuDNN下载地址