Engineering insight

[pytorch 따라하기-1] 구글 Colab에 pytorch 세팅하기 본문

DeepLearning Framework & Coding/Pytorch

[pytorch 따라하기-1] 구글 Colab에 pytorch 세팅하기

Free-Nomad 2021. 7. 25. 19:47

기존에는 저는 Tensorflow라는 Deep Learning Framework을 사용했었는데요,

최신 트렌디한 논문들이나 연구들을 보면 (특히 GAN 관련 연구들이) pytorch의 수가 압도적이라고 느껴질 정도더라구요

 

그래서 이참에 저도 Tensorflow와 Pytorch를 모두 다룰 수 있도록 하기위해

공부하면서 정리해보려고 합니다.

 

 

기본적으로 저는 GAN을 다루기 위해 Pytorch를 사용하는 것이고,

 

집에는 GPU가 없기때문에 Google Colab을 활용하여 무료버전 클라우드 GPU의 개발환경에서 코딩을 진행합니다.

 

Colab사용하기는 하기글을 이용해주세요 :)

https://limitsinx.tistory.com/66 

 

[Google colab-1] 구글 코랩 사용하여 딥러닝 돌리기

보통 딥러닝/머신러닝을 막 입문하신 분들은, GPU를 준비해야하나? RTX, GTX어떤걸 사야하나? 고민이 많으실텐데요 일단 두괄식으로 써드리겠습니다! "이제 막 입문하신분들은 이런것들이 전혀 필

limitsinx.tistory.com

 


Colab에서 Pytorch를 설치하는 방법은 간단합니다.

 

!pip3 install torch

위의 명령어를 실행해주시면 설치가 완료됩니다.

설치하는 방법부터 tensorflow보다 훨씬 간단해보이네요..

 

그럼 다음글부터 본격적으로 pytorch를 다루어보겠습니다!