pytorch gpu 확인 관련

lottopotato / note

skill, reference, method.. note

0 stars 0 forks source link

Open lottopotato opened 10 months ago

lottopotato commented 10 months ago

torch.cuda.is_available()
torch.cuda.device_count()
torch.cuda.current_device()