Win10环境下TensorFlow2.5存在libdevice文件仍提示找不到问题求助

问题原因 #

XLA的NVPTX编译器读取xla_gpu_cuda_data_dir参数时，不会直接在指定目录下搜索libdevice文件，而是会固定拼接/nvvm/libdevice/子路径，再在该子路径下查找对应版本的libdevice文件。你当前将路径直接指向了conda环境的Library/bin目录，就算该目录下存在libdevice.10.bc，XLA实际搜索的是C:/Users/Julian/anaconda3/envs/TF250_PY395_xeus/Library/bin/nvvm/libdevice/libdevice.10.bc，该路径不存在自然会抛出报错。

修复方法 #

方案一（推荐，永久生效） #

• 在conda虚拟环境的Library目录下创建层级文件夹，完整路径为C:/Users/Julian/anaconda3/envs/TF250_PY395_xeus/Library/nvvm/libdevice
• 将C:/Users/Julian/anaconda3/envs/TF250_PY395_xeus/Library/bin目录下的libdevice.10.bc文件，复制到刚才新建的libdevice文件夹中
• 修改XLA_FLAGS参数，指向Library根目录即可，注意该配置代码需要放在所有import TensorFlow的代码之前，否则TensorFlow初始化时会读取旧的环境变量：

import os
os.environ['XLA_FLAGS'] = '--xla_gpu_cuda_data_dir=C:/Users/Julian/anaconda3/envs/TF250_PY395_xeus/Library'

方案二（临时救急） #

直接将libdevice.10.bc文件复制到你当前运行代码的工作目录下，日志显示XLA的搜索路径包含当前目录.，可以直接匹配到报错提示的./libdevice.10.bc路径。

注意事项 #

如果设置系统环境变量后仍不生效，优先检查运行代码的IDE或终端是否加载了最新的环境变量，重启终端/IDE后再测试即可。

内容的提问来源于stack exchange，提问作者Julian Moore