Update app.py
Browse files
app.py
CHANGED
|
@@ -51,6 +51,7 @@ install_cuda_toolkit()
|
|
| 51 |
|
| 52 |
|
| 53 |
print(f"GPU: {torch.cuda.is_available()}")
|
|
|
|
| 54 |
if torch.cuda.is_available() and torch.cuda.device_count() >= 2:
|
| 55 |
device0 = torch.device('cuda:0')
|
| 56 |
device1 = torch.device('cuda:0')
|
|
|
|
| 51 |
|
| 52 |
|
| 53 |
print(f"GPU: {torch.cuda.is_available()}")
|
| 54 |
+
a = torch.tensor([0]).cuda()
|
| 55 |
if torch.cuda.is_available() and torch.cuda.device_count() >= 2:
|
| 56 |
device0 = torch.device('cuda:0')
|
| 57 |
device1 = torch.device('cuda:0')
|