不同的TensorFlow版本路径变了,对于tensorflow-gpu==2.7.0.
用
from tensorflow.python.keras.utils.multi_gpu_utils import multi_gpu_model
替代
from tensorflow.keras.utils import multi_gpu_model
不同的TensorFlow版本路径变了,对于tensorflow-gpu==2.7.0.
用
from tensorflow.python.keras.utils.multi_gpu_utils import multi_gpu_model
替代
from tensorflow.keras.utils import multi_gpu_model