TPU都在运行,然后我激活torch-xla-nightly环境,但当我尝试调用以下简单代码时: import torchimport torch_xla.core.xla_model/site-packages/torch_xla/core/xla_model.py", line 231, in xla_device
devkind=devkind if devkind is not
因为我不是从源代码构建TensorFlow的,所以在默认情况下,XLA搜索文件夹/user/local/cuda-*。但是因为我没有这个文件夹,所以它会发出一个错误。我检查了tensorflow/compiler/xla/service/gpu/nvptx_compiler.cc.中的TensorFlow源代码文件"//用户通过--xla_gpu_ CUDA _data_dir我尝试了以下两个环境变量,但它们都不起作用: expor
我从源代码构建的Tensorflow也是使用XLA支持构建的。我现在还想激活XLA (加速线性代数),因为我希望它能再次提高推理过程中的性能/速度。:CPU for cluster because envvar TF_XLA_FLAGS=--tf_xla_cpu_global_jit was not set.not via TF_XLA_FLAGS) or set the envvar XLA_FLAGS=--xla_hlo_profile.:CPU for