Pytorch Cuda Device Count is 1 but I have 8 GPU cards

1

50
501106×670 95.5 KB

Can somebody help me ?

2

By default pytorch uses only 1 gpu (device:0). If you want to use multiple gpu’s you need to configure it explicitly.

Here is a simple way to do it : https://pytorch.org/tutorials/beginner/blitz/data_parallel_tutorial.html#create-model-and-dataparallel

3

Thanks for your reply. The issue is my torch.cuda.device_count() is 1. However, it should be 8.

4

I’m also getting the same issue
No working solution so far found on the forum :frowning: