Tensorflow (2.0), Keras не использует VRAM от GPU, но частота GPU по-прежнему резко возрастает - PullRequest
1 голос
/ 26 апреля 2020

Я использую сверточную нейронную сеть из курса Deep Learning на Udemy, но когда я это делаю, я вижу, что у моего GPU тактовые импульсы, но процент использования GPU все еще составляет 5% и даже одна эпоха по сравнению с 8000 изображений среднего размера 300 * 400 занимает около 5 минут.

У меня есть Windows 10, RAM - 8 ГБ, GPU - Nvidia Geforce Gtx 1060 6 ГБ

Полный код здесь:

# Convolutional neural network
from keras.models import Sequential
from keras.layers import Conv2D,MaxPooling2D,Flatten,Dense

# Initializing CNN
cl = Sequential()

# Convolution
cl.add(Conv2D(32,3,3, input_shape=(64,64,3),activation='relu'))

# Polling
cl.add(MaxPooling2D(pool_size=(2,2)))

# Flattening
cl.add(Flatten())

# Full Connection
cl.add(Dense(128,activation='relu'))
cl.add(Dense(1,activation='sigmoid'))

# Compiling the CNN
cl.compile(optimizer='adam',loss='binary_crossentropy',metrics=['accuracy'])

# Fitting the CNN

from keras.preprocessing.image import ImageDataGenerator

train_datagen = ImageDataGenerator(rescale = 1./255,
                                   shear_range = 0.2,
                                   zoom_range = 0.2,
                                   horizontal_flip = True)

test_datagen = ImageDataGenerator(rescale = 1./255)

training_set = train_datagen.flow_from_directory('dataset/training_set',
                                             target_size = (64, 64),
                                             batch_size = 100,
                                             class_mode = 'binary')

test_set = test_datagen.flow_from_directory('dataset/test_set',
                                        target_size = (64, 64),
                                        batch_size = 100,
                                        class_mode = 'binary')

cl.fit_generator(training_set,
                     steps_per_epoch = 8000,
                     epochs = 25,
                     validation_data = test_set,
                     validation_steps = 2000)

1 Ответ

0 голосов
/ 26 апреля 2020

То же самое случилось со мной. Если вы загрузите gpu-z, вы увидите, что загрузка графического процессора на самом деле намного выше, чем в диспетчере задач.

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...