머신러닝
Pytorch GPU 사용 여부 확인
pastime
2021. 8. 31. 21:28
728x90
import torch
torch.cuda.get_device_name(0)
torch.cuda.is_available()
torch.__version__728x90