Can PyCharm support two GPUs?

Hello, I have a question. I think I installed CUDA and have been running some projects with it.

print("Number of GPU: ", torch.cuda.device_count())

device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
print('Using device:', device)

output : 

Number of GPU:  1
Using device: cuda


However, I need to know whether the PyCharm platform can support dual GPUs.

Thank you !

0
1 comment

Hi Dimitris, PyCharm fully supports multi-GPU setups for CUDA-based computations. However, PyCharm itself does not control GPU usage, this is managed by your deep learning framework from Python interpreter.

0

Please sign in to leave a comment.