Wie benutze ich mehrere GPUs in Pytorch?Python

Python-Programme
Anonymous
 Wie benutze ich mehrere GPUs in Pytorch?

Post by Anonymous »

Ich benutze diesen Befehl, um eine GPU zu verwenden.

Code: Select all

device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
Aber ich möchte zwei GPUs in Jupyter wie folgt verwenden:

Code: Select all

device = torch.device("cuda:0,1" if torch.cuda.is_available() else "cpu")

Quick Reply

Change Text Case: 
   
  • Similar Topics
    Replies
    Views
    Last post