Tensorflow не работает на процессоре - NHWC - PullRequest
0 голосов
/ 09 апреля 2020

Я получаю это сообщение об ошибке, когда запускаю свой код:

tensorflow.python.framework.errors_impl.InvalidArgumentError: Default AvgPoolingOp only supports NHWC on device type CPU
     [[{{node average_pooling2d/AvgPool}}]]

У меня нет питания с графическим процессором на моем Ma c. Я пытался добавить этот код в начало моего скрипта

import os
import tensorflow as tf

os.environ['CUDA_VISIBLE_DEVICES'] = '-1'

if tf.test.gpu_device_name():
    print('GPU found')
else:
    print("No GPU found")

, печатая «GPU не найден», но я все еще получаю сообщение об ошибке. Я новичок в tenorflow, поэтому мне нужны советы Сом.

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