I faced the same issue when trying to implement the SimCLR framework.
The files are available here
(ssl_models/src at main · sadimanna/ssl_models · GitHub 2)
To reproduce the error I would suggest running ‘python main.py --batch_size 8 --download True’. This will download the CIFAR-10 dataset (163MB).
But when I run the same code on Colab (Google Colaboratory 1) I don’t get the error.
I uploaded the files on Colab and ran the file main.py and got the error again. The notebook with the error is below
(Google Colaboratory 1)
I can’t understand if it is an issue of CUDA or anything else??