INTRO

: Pytorch-Lightning이라고 아는가? Pytorch를 용이하게 쓸 수 있는 라이브러리이다. 최근에 알게 된 라이브러리이다. Computer Vision쪽 AI Researcher 분들 중에서는 Pytorch-Lightning을 사랑하시는 분들이 꽤 계신다고 한다. 그래서 간단한 예제(CIFAR10)를 가지고 Pytorch-Lightning을 Colab에서 여러번 써보다가 "M1에는 설치가 될까?" 라는 생각이 들었고, 그 결과 설치를 해보았고, 그 여정을 공유한다.

Pytorch-Lightning

체감)
1) Pytorch-Lightning으로, 기존 Pytorch 코드를 조금 더 간결하고 용이하게 작성이 가능하다.
2) (나도 Pytorch-Lightning에 빠질까봐 두려울 정도로) 관리 및 사용이 매우 간편하다. 사랑받는 이유가 있긴 하다.
3) 성능상으로 Pytorch와 비슷한 것 같지만, Pytorch 보다 더 뛰어난 것 같지는 않다.
4) Keras, Tensorflow와 Pytorch의 Hybrid 느낌도 있다.
5) Pytorch의 Flow를 알아야 사용이 가능하다. 이게 전제사항이다.

01. Conda Env Clone

: conda 가상환경에 설치할 것이다. Pytorch가 설치된 가상환경이 있다면 clone하여 새로운 가상환경에 Pytorch Lightning을 설치하도록 하자. 당연한 이야기겠지만, Pytorch-Lightning은 Pytorch가 당연히 설치되어있어야한다.

conda create -n jedi --clone night

복사를 했다면, 다음 명령어로 클론하여 만든 가상환경에 진입하자.

conda activate jedi

02. Install Codes at Pytorch-Lightning

: 친절하게도 Apple M1에 설치할 수 있는 코드가 있다. Pytorch-Lightning 페이지에 있다. 코드는 다음과 같다.

export GRPC_PYTHON_BUILD_SYSTEM_OPENSSL=1
export GRPC_PYTHON_BUILD_SYSTEM_ZLIB=1

pip install lightning

03. [Option] Install Lightning-transformers

: 우연히 발견한 것인데, Lightning-transforemrs 라는 라이브러리를 발견했다. 할 짓이 없어서 만든 것 같지만 NLP를 손쉽게 할 수 있도록 만든 라이브러리인 것 같다. 설치 코드는 다음과 같다. 아직 실험을 해보지는 않았다.

pip install lightning-transformers

04 torchtext==0.6

: 왜 인지는 모르겠는데, 위 03 때문인지 모르겠는데, torchtext 버전 관련 에러가 떠서 torchtext 0.6 버전을 설치했다.

pip install torchtext==0.6

이러면 여기까지 설치가 끝난다.
사용에 관한 부분은 나중에 올리겠다.