notebook icon indicating copy to clipboard operation
notebook copied to clipboard

Published 20 hours ago • verified publisher icon jupyter

How to activate 2 GPUs in jupyter notebook ?

Open Iman-Mostafa opened this issue 2 years ago • 1 comments

Iman-Mostafa avatar Mar 21 '22 07:03 Iman-Mostafa

depends on the work you are doing, if you have multiple GPU . . . check it with ! nvidia-smi If you are implementing PyTorch : I would suggest looking into https://pytorch.org/docs/stable/generated/torch.nn.parallel.DistributedDataParallel.html Hope this helps!!!

Prerequisites : check the Cuda package, and if the installation of PyTorch, Tensorflow or others has Cuda version support.

bibhabasumohapatra avatar Sep 02 '22 06:09 bibhabasumohapatra