Ok sure :).
Well I’ve tried both, setting CUDA_VISIBLE_DEVICES=0 or 1 when calling it via CUDA_VISIBLE_DEVICES=0 python script.py and after importing torch via os.environ().
Also as a side note: For whatever reasons, I don’t get a error raise anymore, but go straight to system freezes… it’s really weird