Spaces:
Sleeping
Sleeping
File size: 207 Bytes
12f0268 |
1 2 3 4 5 6 7 8 9 10 11 |
"""
File: gpu_init.py
Author: Elena Ryumina and Dmitry Ryumin
Description: GPU initialization.
License: MIT License
"""
import torch
device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
|