Hi,all
I met the same problem. My cuda version is 10.0 and the python version is 3.6.
I got error message is “RuntimeError: cuda runtime error (11) : invalid argument at /pytorch/aten/src/THC/THCGeneral.cpp:405”
After I used CUDA_LAUNCH_BLOCKING=1 python script.py args, it only used one GPU to work.
Dose anyone have solution?
Merry Christmas!