2023년 9월 12일 화요일

pytorch gpu 사용 여부 확인법

import torch

print(torch.cuda.is_available())

댓글 없음:

댓글 쓰기