인공지능/코드 공부(1)

• [파이썬] pytorch .cuda() 의 쓰임

  다음과 같이 tensor에 device='cuda:0'를 추가해준다

  2022.02.22