Hi guys, I was having the error of THCudaCheck FAIL file=/pytorch/aten/src/THC/THCGeneral.cpp line=383 error=11 : invalid argument and after doing some research around the forum: A error when using GPU 1, I tried doing


CUDA_LAUNCH_BLOCKING=1 python script.py args


torch.backends.cudnn.benchmark = True to False


pip install -U https://download.pytorch.org/whl/cu100/torch-1.0.0-cp36-cp36m-linux_x86_64.whl


All three methods throw the same error even with the first method not showing further details apart from being stuck before the training loop.
I am running on Google Cloud Platform with accelerator="type=nvidia-tesla-t4,count=2". Any help is appreciated. Thank you.