search

LiJunnan1992 / DivideMix

Code for paper: DivideMix: Learning with Noisy Labels as Semi-supervised Learning
MIT License
529 stars 83 forks source link

How to specify gpu #43

Closed kai-wen-yang closed 2 years ago

kai-wen-yang commented 2 years ago

I have 4 gpu on my machine, and I want to run your code on gpu 1, my command is:

python train_cifar.py --gpu_ids 1

But it raises issue:

model = model.to(device)
    RuntimeError: CUDA error: invalid device ordinal

My gpu is normal. Do you have any idea?

LiJunnan1992 commented 2 years ago

The argument is gpuid, singular form.