When I run the test.py in docker, one issue appeared.

Question

When I run the test.py in docker, one issue appeared.

s2244521 opened this issue a year ago · comments

When I run test.py (CSP) in docker, there is an issue appeared.
Traceback (most recent call last):
File "test.py", line 330, in
save_conf=opt.save_conf,
File "test.py", line 55, in test
device = select_device(opt.device, batch_size=batch_size)
File "/yolo/yolov4/utils/torch_utils.py", line 47, in select_device
assert torch.cuda.is_available(), 'CUDA unavailable, invalid device %s requested' % device # check availablity
AssertionError: CUDA unavailable, invalid device 0 requested
I don't know how to solve it, did someone meet this issue?

Qiqiguaiguaide commented a year ago

+1

s2244521 · Answer 1 · Thu Mar 23 2023 15:32:41 GMT+0800 (China Standard Time)

hi @Qiqiguaiguaide
I solved this problem, I changed the device to cpu and it works. I think it because my CUDA version and pytorch version can not match the environment setting of this code. My CUDA version is 12.