Num_gpu <= torch.cuda.device_count() Assertion error

Hello there,
I meet an issue which shows above


However as I search through the internet I couldn’t find any answer for that. Does anyone have any idea what’s the issues for that and give me any solution? Thanks!!

How many GPUs do you have?
num gpus in your code should be less than the number of GPUs you have

From terminal it says only one gpu available, as I’m implementing git from

From github it doesn’t specify gpu num

In argument, num gpus is set to 8.

$ python xxx.py --num-gpus 1

Let’s see how this change