티스토리 뷰

728x90
반응형

개발에 있어서 개발 환경은 매우 중요한 요소이다.

딥러닝 모델을 구성할 때도 pytorch 버전에 따라 명령어가 차이가 나기 때문에 현재 진행하는 프로젝트에 어떤 패키지들이 어떤 버전으로 설치되어 있는지 기록해 두는 것이 중요하다.

requirements.txt 파일은 현재 개발 환경(python)에 pip를 통해 설치된 모든 패키지 목록이 버전과 함께 기록되는 곳이다.

1. requirements.txt 생성 명령어

requirements.txt 파일을 만드는 방법은 두 가지가 있다.

  1. pip freeze 명령어를 사용하는 법
  2. pip list 명령어를 사용하는 법

  1.1. pip freeze 명령어

pip freeze > requirements.txt

pip freeze를 사용하여 requirements.txt 파일을 만들면 아래와 같이 패키지의 다운로드 주소가 @으로 표시되기도 한다.

torch @ https://download.pytorch.org/whl/cu116/torch-1.13.1%2Bcu116-cp38-cp38-linux_x86_64.whl
torchaudio @ https://download.pytorch.org/whl/cu116/torchaudio-0.13.1%2Bcu116-cp38-cp38-linux_x86_64.whl
torchsummary==1.5.1
torchtext==0.14.1
torchvision @ https://download.pytorch.org/whl/cu116/torchvision-0.14.1%2Bcu116-cp38-cp38-linux_x86_64.whl
tqdm==4.64.1

  1.2. pip list 명령어

pip list --format=freeze > requirements.txt

pip list를 사용하면 아래와 같이 requirements.txt 파일에는 패키지 이름과 버전만이 기록이 된다.

torch==1.13.1+cu116
torchaudio==0.13.1+cu116
torchsummary==1.5.1
torchtext==0.14.1
torchvision==0.14.1+cu116
tqdm==4.64.1

2. requirements.txt 설치 명령어

2.1. pip install 명령어

requirements.txt에 적힌 대로 버전에 맞게 모든 패키지들을 설치한다.

pip install -r requirements.txt

< 주의 >

리눅스에서 설치한 패키지 목록은 윈도우에서 설치가 안 될 수 있으니 리눅스에서 시도해 보면 좋을 것 같다.

728x90
반응형
댓글