TensorFlow、PyTorch各版本对应的CUDA、cuDNN关系

TensorFlow、PyTorch各版本对应的CUDA、cuDNN关系(截止2021年4月7日)

  • TensorFow
    • Linux
      • CPU
      • GPU
    • macOS
      • CPU
      • GPU
    • Windows
      • CPU
      • GPU
  • PyTorch
    • CPU
    • GPU
  • cuDNN与CUDA
  • CUDA和NVIDIA显卡驱动关系

因为之前tensorflow、pytorch配环境在cuda上吃过很多亏,主要是版本不匹配造成的,所以自行整理了对应关系,资料均来自官网。

TensorFow

官网:https://www.tensorflow.org/?hl=zh-cn

Linux

CPU

TensorFlow、PyTorch各版本对应的CUDA、cuDNN关系_第1张图片

GPU

TensorFlow、PyTorch各版本对应的CUDA、cuDNN关系_第2张图片

macOS

CPU

TensorFlow、PyTorch各版本对应的CUDA、cuDNN关系_第3张图片

GPU

TensorFlow、PyTorch各版本对应的CUDA、cuDNN关系_第4张图片

Windows

CPU

TensorFlow、PyTorch各版本对应的CUDA、cuDNN关系_第5张图片

GPU

TensorFlow、PyTorch各版本对应的CUDA、cuDNN关系_第6张图片

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
TensorFlow、PyTorch各版本对应的CUDA、cuDNN关系_第7张图片

你可能感兴趣的:(深度学习)