我刚刚将操作系统更改为Ubuntu，但我没有设法在GPU上使用TF。在Windows上工作正常。
我有一个Nvida MX150，通过conda安装在python 3.6.8和tensorflow-gpu 1.12上运行。
当我尝试使用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工作正常。有人遇到过类似的问题吗？