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

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

因为之前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

PyTorchtorchvision
1.8.00.9.0
1.7.10.8.2
1.7.00.8.0
1.6.00.7.0
1.5.10.6.1
1.5.00.6.0
1.4.00.5.0
1.2.00.4.0
1.1.00.3.0
1.0.10.2.2
1.0.00.2.1

GPU

PyTorchCUDAtorchvision
1.8.010.2、11.10.9.0
1.7.19.2、10.1、10.2、11.00.8.2
1.7.09.2、10.1、10.2、11.00.8.0
1.6.09.2、10.1、10.20.7.0
1.5.19.2、10.1、10.20.6.1
1.5.09.2、10.1、10.20.6.0
1.4.09.2、10.10.5.0
1.2.09.2、10.00.4.0
1.1.09.0、10.00.3.0
1.0.19.0、10.00.2.2
1.0.08.0、9.0、10.00.2.1

cuDNN与CUDA

官网:https://developer.nvidia.com/rdp/cudnn-archive

cuDNNCUDA
8.1.011.2、11.1、11.0、10.2
8.0.511.1、11.0、10.2、10.1
8.0.411.1、11.0、10.2、10.1
8.0.311.0、10.2、10.1
8.0.211.0、10.2、10.1
8.0.111.0、10.2
7.6.510.2、10.1、10.0、9.2、9.0
7.6.410.1、10.0、9.2、9.0
7.6.310.1、10.0、9.2、9.0
7.6.210.1、10.0、9.2、9.0
7.6.110.1、10.0、9.2、9.0
7.6.010.1、10.0、9.2、9.0
7.5.110.1、10.0、9.2、9.0
7.5.010.1、10.0、9.2、9.0
7.4.210.0、9.2、9.0
7.4.110.0、9.2、9.0
7.4.010.0、9.2、9.0
7.3.010.0、9.0
7.2.19.2

CUDA和NVIDIA显卡驱动关系

官网:https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html
在这里插入图片描述

  • 61
    点赞
  • 354
    收藏
    觉得还不错? 一键收藏
  • 6
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 6
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值