you must’ve tried to do:
CUDA_VISIBLE_DEVICES=4 python example.py
CUDA_VISIBLE_DEVICES is 0-indexed, so if you have 4 GPUs, then valid values are 0, 1, 2 or 3.
pytorch (actually CUDA) is complaining that you gave an invalid device id to use