Torch.cuda.device_count() return 1 with 1080Ti GPU

1

i have a 1080Ti GPU with ubuntu os. but when i run the command line

torch.cuda.device_count()

it returns 1. i also try it in terminal, it also return 1
i google this problem, find this
but this link it provides doesn’t work

image
image795×638 92.8 KB

2

Hi,

What is the problem? If you have one GPU, then the device count is 1.

1 Like
3

thank you, I confused the two concepts of GPU and GPU computing unit.
i thought the return of this command is the count of GPU computing unit. thank you