1. ホーム
  2. テンソルフロー

tensorflow CUDA_ERROR_ILLEGAL_ADDRESSの解決法(Windows版)。

2022-03-01 11:16:22

最近、tensorflowで以下のようなバグが発生しました。

2017-11-08 12:24:52.838039: [E tensorflow/stream_executor/cuda/cuda_driver.cc:1080] failed to synchronize the stop event: CUDA_ERROR_ILLEGAL_ ADDRESS
2017-11-08 12:24:52.838090: E tensorflow/stream_executor/cuda/cuda_timer.cc:54] Internal: error destroying CUDA event in context 0x51f18f0: CUDA_ERROR_ILLEGAL_ADDRESS
2017-11-08 12:24:52.838106: E tensorflow/stream_executor/cuda/cuda_timer.cc:59] Internal: error destroying CUDA event in context 0x51f18f0: CUDA_ERROR_ILLEGAL_ADDRESS
2017-11-08 12:24:52.838137: [F tensorflow/stream_executor/cuda/cuda_dnn.cc:3218] failed to set stream for cudnn handle: CUDNN_STATUS_MAPPING_ ERROR

として https://github.com/tensorflow/tensorflow/issues/14363 のように、cudnn 7.0.5を後から入れても解決しませんでした。

私のモデルは tf.contrib.image. <スパン トランスフォーム のレイヤーは、ubuntuのgpu版は問題なく、windowsのcpu版は問題ないが、gpu版はバグがある。 https://github.com/tensorflow/tensorflow/issues/17485 図に示すように tf.contrib.image. トランスフォーム cpuに載せて、読む。

with tf.device('/cpu:0'):
    data_node_transformed = tf.contrib.image.transform(imgs, rtm, "BILINEAR")

モデルを再トレーニングし、デプロイします。


その後のtensorflowとnvidiaによるバグフィックスを期待します。