Spaces:

lablab-ai-amd-developer-hackathon
/

QATALY_STS_DS

Sleeping

QATALY_STS_DS / src /qatalyst /shared.py

initial public demo release

ddb21b6 6 months ago

157 Bytes

import torch

DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
DTYPE = torch.float32

torch.set_default_device(DEVICE)
torch.set_default_dtype(DTYPE)

	import torch

	DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
	DTYPE = torch.float32

	torch.set_default_device(DEVICE)
	torch.set_default_dtype(DTYPE)