Tensorflow-Directmlで`DXGI_ERROR_DEVICE_REMOVED`が出てしまうときの対処法 - 端の知識の備忘録

概要 前回もご紹介したTensorflow-Directmlに関するエラーについて。 この間久しぶりにTensorflow-Directmlを使ったところ、DXGI_ERROR_DEVICE_REMOVEDというエラーが出て学習前にコケてしまう事態に見舞われた。 Tensorflow-Directml の日本語記事はあまりないので、とりあえずメモ。 まとめ 確認した環境は、RX6900XT, python3.7にtensorflow-directmlの各バージョン。 1.15.5.dev210429, 1.15.5.dev210422, 1.15.4.dev201216, 1.15.3.dev200911 といろいろバージョンを試してみたが同じエラーが出てしまった。 対処法は、DMLデバイスを使う前に予め環境変数TF_DIRECTML_MAX_ALLOC_SIZEで割り当てるメモリの量を決