快速配置tensorflow-gpu版本

1)首先确认自己安装的系统驱动版本,在terminal中使用nvcc -V
快速配置tensorflow-gpu版本_第1张图片
这里看到使用的cuda版本是11.3

2)从推荐的配置单中查看tensorflow版本、cudnn版本和cuda版本之间的关系;
tf2.6/tf2.7 cuda 11.3 cudnn 8.2
3) 确认使用的版本之后,在terminal中输入: conda install cudnn=8.2

4)等待3执行完之后,2)中的推荐配置就能对的起来了,这时候直接使用就OK了。
快速配置tensorflow-gpu版本_第2张图片

你可能感兴趣的:(问题解决,tensorflow,python,深度学习)