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