Я только что сменил ОС на Ubuntu, и мне не удается использовать TF на моем GPU.Это работало очень хорошо на окнах.У меня есть Nvida MX150, работающий на python 3.6.8 и tenorflow-gpu 1.12, из установки conda.Когда я пытаюсь использовать device=GPU:0, я получаю сообщение:
Cannot assign a device for operation Shape: Operation was explicitly assigned to /device:GPU:0 but available devices are [ /job:localhost/replica:0/task:0/device:CPU:0, /job:localhost/replica:0/task:0/device:XLA_CPU:0, /job:localhost/replica:0/task:0/device:XLA_GPU:0 ].
Но если я попробую XLA_GPU, ячейка jupyter никогда не завершит свою работу.XLA_CPU работает просто отлично.Кто-нибудь сталкивался с подобной проблемой?