search

MenghaoGuo / AutoDeeplab

Pytorch Implementation the paper Auto-DeepLab Hierarchical Neural Architecture Search for Semantic Image Segmentation
https://arxiv.org/abs/1901.02985
410 stars 97 forks source link

Single GPU with workers=1, but I still met cuda runtime error. #23

Closed ShunLu91 closed 5 years ago

ShunLu91 commented 5 years ago

I use a single GPU card( NVIDIA GTX1080 Ti ) and set the workers to 1 according to other issues. However, when I ran the train_autodeeplab.py, after some images, I still met the problem that cuda runtime error (2): at c:\programdata\miniconda3\conda-bld\pytorch_1524549877902\work\aten\src\thc\generic/THCStorage.cu:58. Could somebody tell me how to sovle this if I only have a single GPU card? image

ShunLu91 commented 5 years ago

I have solved this problem by follow #14 like this image