Ubuntu18解决:Not creating XLA devices, tf_xla_enable_xla_devices not set
实验环境
ubuntu18
NVIDIA GeForce GTX 1050
CUDA11.0
CUDNN 8.0
Tesorflow-gpu 2.4.1
问题
分析原因
据说是Tesorflow-gpu 2.4.1,默认情况下,不再注册XLA:CPU和XLA:GPU设备。如果确实需要它们,请可使用
os.environ['TF_XLA_FLAGS'] = '--tf_xla_enable_xla_devices'
但是此标志最终将在后续发行版中删除。
解决方法
将
import tensorflow
原创
2021-05-07 22:05:00 ·
536 阅读 ·
0 评论