megstudio是不支持tensorflow吗？为什么读不出GPU？

#1

在gpu环境下，pytorch正常。

但输入一下tensorflow官方的，获取gpu信息的代码

import tensorflow as tf
print("Num GPUs Available: ", len(tf.config.experimental.list_physical_devices(‘GPU’)))

返回是0
tensorflow获取不到GPU是什么原因?

#2

收到反馈，我们证在排查原因

#3

这个问题应该是你装的 tensorflow 和 cuda 版本不兼容的问题，可以进入环境后开启终端，通过命令 “cat /usr/local/cuda/version.txt” 查看环境的 cuda 版本，然后pip 安装对应版本的 tf - gpu。