https://github.com/baowenbo/DAIN のコードをcolaboratoryで実行しようとして、
期待通りに demo_MiddleBury.py の実行が出来、期待通りの出力も得られました。
実行手順はgithubのREADMEの通りにしましたが、
元のdemo_MiddleBury では
from scipy.misc import imread, imsave
となっている所を、以下に変えていました。
from imageio import imread, imsave
2月7日より後の段階では、同じ手順を踏んでも、
○RuntimeError: CUDA call failed
○error in correlation_forward_cuda_kernel:
等のエラーで実行が停止します。
全く同じ手続きをしていると思っているのですが、何がおかしいのでしょうか?
もしかしたら、数日前の段階では何かを設定していて、私がそれを忘れているのかもしれないのですが
colaboratory で DAIN のコードは実行出来るのでしょうか?
実行したことが無いのでよくわかりませんが、ランタイムのタイプ変更を忘れていないかを確認し、後は https://github.com/baowenbo/DAIN/issues/44 に少し情報がありますね。最後の人が以下にjupyterのコードを上げてくれています。
https://colab.research.google.com/drive/1hkPQQNRH1ykKJN6V7tTiGnZOHWm7JRdr
他にはcolaboとDAINの環境のCudaのバージョンが合わなくなったのかもしれませんね。
CUDAのバージョン:!nvcc --version
cudnnのバージョン:!ls -l /usr/lib/x86_64-linux-gnu/libcudnn.so*
ドライバーのバージョン:! ls /usr/lib64-nvidia/libcuda*
実行したことが無いのでよくわかりませんが、ランタイムのタイプ変更を忘れていないかを確認し、後は https://github.com/baowenbo/DAIN/issues/44 に少し情報がありますね。最後の人が以下にjupyterのコードを上げてくれています。
https://colab.research.google.com/drive/1hkPQQNRH1ykKJN6V7tTiGnZOHWm7JRdr
他にはcolaboとDAINの環境のCudaのバージョンが合わなくなったのかもしれませんね。
CUDAのバージョン:!nvcc --version
cudnnのバージョン:!ls -l /usr/lib/x86_64-linux-gnu/libcudnn.so*
ドライバーのバージョン:! ls /usr/lib64-nvidia/libcuda*
最後の人のコードは「そのまま」実行しましたが、
まったく同じエラーでした
ランタイプを何に変更するのでしょうか?
python3 GPUになっています
Colabの動作環境が新しくなった可能性が高いように思います。
現在のColabの環境を調べてみるとUbuntu18.04.3、CUDA10になっているので、DAINの環境よりも新しくなってます。CUDA9にダウンさせることができるかというと、CUDA9.2が対応しているのがUbuntu17.10と16.04なので、OSもダウンさせないといけないかもせれません。CUDAはともかくOSは入れ替えることができないのではないかと思いますので、GitHubでCUDA10に対応するようにRequestするのが一番早いのかと思います。もしくはご自身でDAINのコードを理解してCUDA10用に修正する必要があるでしょう。Colabで動かないというIssue以外にもCUDA10.2で動かないというIssueも挙がっているので、CUDAのバージョンが合わない為に動かないという事で、間違いないと思います。
最後の人のコードは「そのまま」実行しましたが、
2020/02/12 12:17:48まったく同じエラーでした
ランタイプを何に変更するのでしょうか?
2020/02/12 13:00:01python3 GPUになっています