日々精進

新しく学んだことを書き留めていきます

PyTorchを実行すると「ctdet/coco_res18THCudaCheck FAIL file=/opt/conda/conda-bld/pytorch_1535491974311/work/aten/src/THC/THCGeneral.cpp line=663 error=11 : invalid argument」エラー

PyTorchのバージョンが古いと発生するらしい。使っていたバージョンは0.4.0。 以下サイトでinstallコマンドがわかるので、これを実行して最新版をインストールしたら直った。

pytorch.org

参考:

discuss.pytorch.org

discuss.pytorch.org