Win10中文家庭版+RTX3060配置Tensorflow的GPU开发环境

不知道痛下血本,购进NVIDIA RTX系列显卡,准备做深度学习训练的朋友有没有碰到开发环境配置上的困境?

这是因为RTX系列显卡需要CUDA11.x支持,而目前Anconda中,conda命令默认安装的TensorFlow-gpu版本是2.3.0,它不支持CUDA11。CUDA11需要安装2.5.0及以上版本TensorFlow。

而且特别要注意的是TensorFlowGPU环境运行的稳定与否与tf、CUDA和cuDNN三者间版本匹配的程度直接相关,图下图所示:
Win10中文家庭版+RTX3060配置Tensorflow的GPU开发环境_第1张图片
所以目前最推荐的搭配是CUDA11.2 + cuDNN8.1 + tensorflow_gpu-2.5.0

你可能感兴趣的:(环境配置,tensorflow,深度学习,机器学习)