#python #tensorflow
#python #тензорный поток
Вопрос:
Я использую python3.6.8, TensorFlow-GPU 2.0.0 nv, CUDA V10.2.89
Мой графический процессор 
Я хочу использовать только 1 графическое устройство, поэтому я использовал этот код для добавления видимого графического устройства в Tensorflow
import os
os.environ["CUDA_VISIBLE_DEVICES"] = "7"
os.environ['TF_FORCE_GPU_ALLOW_GROWTH'] = 'true'
config = tf.compat.v1.ConfigProto(allow_soft_placement=True, log_device_placement=True)
config.gpu_options.allow_growth = True
tf.compat.v1.Session(config=config)
Но когда я запускаю код, все еще вижу другие устройства GPU
И, наконец, ошибка, указанная в cuDNN, не удалось инициализировать 
Но когда я решаю использовать графический процессор с номером 0, все в порядке, ошибок нет!!! (Тот же код и та же среда просто меняют только номер устройства GPU с 0 на 7) 
PS Я использую изображение Docker от Nvidia (nvcr.io/nvidia/tensorflow ), Поэтому я думаю, что все совместимо (?)
Комментарии:
1. Похоже, вы импортировали tensorflow перед настройкой
CUDA_VISIBLE_DEVICES, это правильно? Возможно, вам придется сделать это наоборот.2. @myrtlecat Большое вам спасибо! Я вызвал TF перед настройкой моей ошибки: (