Something doesn’t seem right.
The _check_balance method only checks for the GPU specs (total memory and multi processor count).
So if all cards are 1080Tis the warning shouldn’t come.
In your first attempt to use DataParallel you used device_ids=[0, 3] and got “Invalid device id” back.
Are you sure you are working on the right server?