cuda
Browse files
main.py
CHANGED
|
@@ -22,6 +22,7 @@ import uuid
|
|
| 22 |
import numpy as np
|
| 23 |
import cv2
|
| 24 |
|
|
|
|
| 25 |
# Use GPU if available
|
| 26 |
if torch.cuda.is_available():
|
| 27 |
device = torch.device("cuda")
|
|
|
|
| 22 |
import numpy as np
|
| 23 |
import cv2
|
| 24 |
|
| 25 |
+
print(torch.__version__)
|
| 26 |
# Use GPU if available
|
| 27 |
if torch.cuda.is_available():
|
| 28 |
device = torch.device("cuda")
|