본문 바로가기

TIL/개인공부

[Python] whl파일 생성시 setup.py install_requires 특정 파일로 install되게끔 설정하는 방법

반응형

업무중에 패키징화(whl파일생성)를 하는 작업이있는데, install_requires부분에 특정파일을 이용하여 설치가 필요한 패키지가있었다.

사실 패키징화만 아니라면 간단하게 아래 명령어로 설치할수있다.

pip install {PACKAGE_NAME} -f {PACKAGE_URL or PACKAGE_FILE_LOCATION}

 

하지만 setup.py의 install_requires에는 위와같은 -f 옵션을 별도로 넣어줄수가없는 상태여서 설치가 어렵다.

여러 사이트를 조사해본결과 아래와같이 해결할수있다.

 

install_requires=[
                "{PACKAGE_NAME}@{PACKAGE_URL}"
                ],

패키지 이름과 패키지 위치 또는 URL정보 사이에 골뱅이(@)를 넣어주면 해결된다! 아래는 torchvision에 대한 예시이다.

install_requires=[
    "torchvision@https://download.pytorch.org/whl/cu110/torchvision-0.8.2%2Bcu110-cp38-cp38-linux_x86_64.whl",
    ],

 

반응형