전공 - AI/딥러닝
Google Colab 에서 torch 사용하는 방법
잉카콜라
2023. 4. 1. 22:00
구글 Colab 에서 torch 를 사용하기 위해 기초 설정을 해 주었다

우선 torch 관련 파일을 받은뒤 실행하지 않은 상태이다

위쪽에 런타임을 클릭하고 런타임 유형 변경을 클릭한다

하드웨어 가속기가 None 으로 되어있을 건데 GPU로 바꿔주고 저장을 누른 뒤 실행하면 정상적으로 실행이 된다

728x90