mirror of
https://github.com/BoardWare-Genius/jarvis-models.git
synced 2025-12-13 16:53:24 +00:00
5 lines
108 B
Python
5 lines
108 B
Python
import torch
|
|
|
|
print("Torch version:",torch.__version__)
|
|
|
|
print("Is CUDA enabled?",torch.cuda.is_available()) |