1、使用anaconda安装tensorflow-gpu
conda install tensorflow-gpu
conda install -c anaconda cudatoolkit
conda install -c anaconda cudnn
2、Not creating XLA devices, tf_xla_enable_xla_devices not set
解决办法
实际上这个问题可以忽略,看看2.4版本的release就一目了然,并不是很多博客说的版本对应问题,回退到老版本治标不治本。
如果需要用XLA,添加TF_XLA_FLAGS=–tf_xla_enable_xla_devices即可解决该warning。
release
默认情况下,不再注册XLA:CPU和XLA:GPU设备。
TF_XLA_FLAGS=--tf_xla_enable_xla_devices
如果确实需要它们,请使用此flag,但是此flag最终将在后续版本中删除。