有许多文章试图在 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] 类似问题列表