关于NVIDIA A100-sxm4-40搭配CUDA11.0、CuDNN8.0适配TensorFlow版本的咨询
适配CUDA 11.0、CuDNN 8.0与A100 GPU的TensorFlow版本
- 针对你的配置(CUDA 11.0 + CuDNN 8.0 + NVIDIA A100-sxm4-40 GPU),最新的适配TensorFlow稳定版本是2.5.x系列(推荐安装2.5.3,该版本是2.5分支的最终稳定版)。
- 适配说明:
- TensorFlow 2.4.x到2.5.x官方明确支持CUDA 11.0和CuDNN 8.0;
- A100基于Ampere架构,TensorFlow 2.4及以上版本对该架构GPU提供完整支持;
- TensorFlow 2.6及以上版本最低要求CUDA 11.2,无法直接适配你的CUDA 11.0环境。
内容的提问来源于stack exchange,提问作者THEODOROPOULOS DIMITRIS
相关产品推荐
相关产品推荐

