Google Colab環境でRuntimeError: cuda runtime error (30)と表示される

Question

Google Colabで実行しています。
つい先ほどまで実行に成功していたのですが、突然エラーが表示されるようになりました。

!git clone https://github.com/t-Asai/CartoonGAN-Test-Pytorch-Torch.git
%cd CartoonGAN-Test-Pytorch-Torch
!pip install -r requirement.txt
!sh pretrained_model/download_pth.sh

でCartoonGANをgit cloneして、訓練済みモデルを使って以下のコードで実行すると

!python test.py --input_dir image_dir --output_dir out_dir --style Hayao

以下のエラーが表示されます。

GPU mode
THCudaCheck FAIL file=/pytorch/aten/src/THC/THCGeneral.cpp line=74 error=30 : unknown error
Traceback (most recent call last):
  File "test.py", line 142, in <module>
    main()
  File "test.py", line 95, in main
    model.cuda()
  File "/usr/local/lib/python3.6/dist-packages/torch/nn/modules/module.py", line 258, in cuda
    return self._apply(lambda t: t.cuda(device))
  File "/usr/local/lib/python3.6/dist-packages/torch/nn/modules/module.py", line 185, in _apply
    module._apply(fn)
  File "/usr/local/lib/python3.6/dist-packages/torch/nn/modules/module.py", line 191, in _apply
    param.data = fn(param.data)
  File "/usr/local/lib/python3.6/dist-packages/torch/nn/modules/module.py", line 258, in <lambda>
    return self._apply(lambda t: t.cuda(device))
RuntimeError: cuda runtime error (30) : unknown error at /pytorch/aten/src/THC/THCGeneral.cpp:74

自分で調べてはみたのですが、参考になるような情報が得られず、またローカル環境でないため勝手がわからず手に負えません。どなたかお力添えいただけると幸いです。

追記

sudo pythonも試しましたがダメでした。

Comments

[tiitoi]

経験上、unkown Error はメモリ不足のときに起こる気がします。
Google Colab は使ったことがないのですが、GPU ってたしか共有でしたよね？
他にも利用者がいて、GPU が利用中とかではないでしょうか？
!nvidia-smi で使用状況を確認してみてください

Answer 1

ここにあるように--cuda 0を指定してはどうでしょう．
最後に成功してからエラーが出るまでなにか操作はされましたか？