tensorflow - Ubuntu下RTX 30上的Conda（Tensorflow，Keras）->编译时间很长

Question

有许多文章试图在 RTX 30 [1] 下配置 Tensorflow，但很少（如果有的话）成功。

版本矩阵：

• Ubuntu 20.04.03 LTS
• 张量流：2.4.1
• NVIDIA-SMI 495.29.05
• 司机：495.29.05
• CUDA：11.5
• cuDNN：8.3.0.98-1

下面的一个简单片段需要 5-6 分钟来编译/输出：

from tensorflow.python.client import device_lib
print(device_lib.list_local_devices())

[1] 类似问题列表

• 为什么 Tensorflow GPU 在创建模型和训练模型时与 CPU 版本相比速度极慢？
• 使用 tensorflow keras 配置 RTX 3090
• https://github.com/tensorflow/tensorflow/issues/48787
• https://github.com/tensorflow/tensorflow/issues/44936
• https://github.com/tensorflow/tensorflow/issues/48560

标签： tensorflowkerasanacondartx