

pytorch を用いて作成したモデルから計算された損失を  
伝搬する際に、以下のコードを実行すると  
以下のエラーメッセージが表示されて困っています。

[code]

    loss.backward()
    
[/code]

順伝搬の計算は問題なく実行できます。

[code]

    terminate called after throwing an instance of 'std::runtime_error'
      what():  tensorflow/compiler/xla/xla_client/computation_client.cc:280 : Missing XLA configuration
    Aborted
    
[/code]

  * pytorch(1.12.0+cu102)
  * torchvision(0.13.0+cu102) <\- ターゲットのモデルは torchvision.models からインストールできる CNN モデルを含んでいます
  * google-compute-engine
  * GPU(NVIDIA Tesla T4 x 1, 11.6) <\- GPU(11.2)がインストールされている環境では動いていたコードなのですが、現在の環境では動かないです・・・ / 現在の環境では、GPU を利用せず、CPU で動かしても同じエラーが出ます
  * TPU はインストールされていません(TPU を使いたいわけではなく、GPUで計算させたいです)

ローカルでは動いているコードで、上述した通り他の GPU 環境でも動いていました。環境がアップデートされたタイミングで動かなくなりました。

助けてください・・・


CC BY-SA 4.0
2022-10-12T08:05:45.807
91562
2022-10-13T06:17:12.380
2022-10-12T21:30:42.123
54809
54809
question
python pytorch gpu
loss.backward() -> Missing XLA configuration


以下のコマンドで解決しました。

[code]

    $ pip uninstall torch_xla
    
[/code]

pytorch-ignite と torch_xla の問題であったようです。


CC BY-SA 4.0
2022-10-13T06:17:12.380
91577
2022-10-13T06:17:12.380
54809
91562
answer