I have a laptop with Nvidia 940MX graphics card with 4GB memory, so I thought I could run the code in notebooks locally. I have anaconda Python 3.6 + PyTorch + CUDA+CUDNN installed.
When I try to run the lesson-1.ipynb notebook (dogs vs cats), my browser (and laptop) hangs at the line “
ConvLearner.pretrained(resnet34, data, precompute=True)” step. I see a message “Web socket timed out after 111585 ms” in the terminal. I used different browsers (Chrome, FireFox), but it still persists. I have tried using the following code, but I still see it:
To check whether my GPU is being used, I used “nvidia-smi” in the terminal, and it shows the process “~/anaconda3/python” using 1289 MB, so I guess the code is NOT being run by CPU.
The dogs vs cats runs fine with a small sample dir (each 10 images), but I’m unable to run with full data, probably due to the websocket time out issue. Can someone help me with this?