我正在尝试运行以下代码行:
print("Num GPUs Available: " , len(tensorflow.config.experimental.list_physical_devices('GPU')))但它返回错误:
AttributeError: module 'tensorflow' has no attribute 'config'你知道我做错了什么吗?
编辑:这是初始化代码
from __future__ import absolute_import, division, print_function, unicode_literals
import os
from glob import glob
import time
import tensorflow
#os.environ['KERAS_BACKEND'] = 'tensorflow'
#os.environ['TF_CPP_MIN_LOG_LEVEL'] = '3' # 3 = INFO, WARNING, and ERROR messages are not printed
from tqdm import tqdm
from keras.utils import np_utils
import numpy as np
import pandas as pd
from IPython.display import FileLink
import matplotlib.pyplot as plt
import warnings
warnings.filterwarnings('ignore')
import seaborn as sns
from IPython.display import display, Image
import matplotlib.image as mpimg
import cv2
from sklearn.model_selection import train_test_split
from sklearn.datasets import load_files
from sklearn.utils import shuffle
from sklearn.metrics import log_loss
from keras.applications.vgg16 import VGG16
from keras.models import Sequential, Model
from keras.layers import Conv2D, MaxPooling2D, Flatten, Dense, Dropout, BatchNormalization, GlobalAveragePooling2D, Input
from keras.preprocessing.image import ImageDataGenerator
from keras.preprocessing import image
from keras.callbacks import ModelCheckpoint, EarlyStopping
import tflearn
from tflearn.layers.conv import conv_2d, max_pool_2d
from tflearn.layers.core import input_data, dropout, fully_connected
from tflearn.layers.estimator import regression
from tensorflow.keras import datasets, layers, models我也在使用Tensorflow 2.0。如果有人能帮上忙那就太好了
发布于 2020-09-06 07:32:36
对于TensorFlow2.x,您应该使用tf.test.gpu_device_name()
import tensorflow as tf
print("Num of GPUs available: ", len(tf.test.gpu_device_name()))
Num of GPUs available: 13发布于 2020-02-02 08:13:27
这可能是因为您已经安装了TensorFlow2.0,但代码是为TensorFlow1.0编写的
AttributeError: module 'tensorflow' has no attribute 'ConfigProto'
请试一下,看看是否有效。
https://stackoverflow.com/questions/60022195
复制相似问题