我的系统有Tensorflow 2.2 (Python 3.8, anaconda)
. 我阅读了有关 Tensorflow XLA 编译器的信息,据我了解,它是 Tensorflow 附带的预编译 GPU 操作的替代方法。即使我在没有急切执行的情况下运行代码或者experimental_compile=True
我XLA service 0x5644b1cec990 initialized for platform CUDA (this does not guarantee that XLA will be used)
在终端中看到。此外,如果它以某种方式链接,则Executing op Fill in device /job:localhost/replica:0/task:0/device:GPU:0
在进行下一个操作之前需要相当长的时间(约 2 分钟)。
问问题
540 次