전공 - AI/딥러닝

Google Colab 에서 torch 사용하는 방법

잉카콜라 2023. 4. 1. 22:00

구글 Colab 에서 torch 를 사용하기 위해 기초 설정을 해 주었다

 

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

 

 

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

 

 

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

 

728x90