xcx0902
/

Multiplication

Model card Files Files and versions

Multiplication / device.py

xcx0902's picture

Upload folder using huggingface_hub

2eebda5 verified 11 months ago

140 Bytes

	import torch

	device: str

	if torch.cuda.is_available(): device = 'cuda'
	elif torch.mps.is_available(): device = 'mps'
	else: device = 'cpu'