본문 바로가기
DeepLearning Framework & Coding/Pytorch

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

by 노마드공학자 2021. 7. 25.

기존에는 저는 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를 다루어보겠습니다!

댓글