不知道痛下血本,购进NVIDIA RTX系列显卡,准备做深度学习训练的朋友有没有碰到开发环境配置上的困境?
这是因为RTX系列显卡需要CUDA11.x支持,而目前Anconda中,conda命令默认安装的TensorFlow-gpu版本是2.3.0,它不支持CUDA11。CUDA11需要安装2.5.0及以上版本TensorFlow。
而且特别要注意的是TensorFlowGPU环境运行的稳定与否与tf、CUDA和cuDNN三者间版本匹配的程度直接相关,图下图所示:
所以目前最推荐的搭配是CUDA11.2 + cuDNN8.1 + tensorflow_gpu-2.5.0