simple-faster-rcnn-pytorch icon indicating copy to clipboard operation
simple-faster-rcnn-pytorch copied to clipboard

Published 20 hours ago • verified publisher icon chenyuntc

Can I use another GPU on my computer?

Open Flawless1202 opened this issue 6 years ago • 2 comments

I want to use another GPU on my computer, but when I change all .cuda() to .cuda(2) this error occured: cupy.cuda.driver.CUDADriverError: CUDA_ERROR_INVALID_HANDLE: invalid resource handle What can I do to solve this?

Flawless1202 avatar Sep 09 '18 02:09 Flawless1202

Try running the code by adding os.environ['CUDA_VISIBLE_DEVICES'] = '2' in train.py or CUDA_VISIBLE_DEVICES = 2 python train.py xxxxxx

Hope this might help?

rxqy avatar Sep 26 '18 08:09 rxqy

I got the same problem, setting os.environ['CUDA_VISIBLE_DEVICES'] doesn't work! have u solved it now?

chrisway613 avatar Mar 31 '20 10:03 chrisway613