FuriosaAI Python SDK 설치

FuriosaAI NPU Python SDK는 NPU 에서 모델을 가속시키기 위한 각종 인터페이스를 제공하며, Python 라이브러리 및 명령 줄 도구(command line tool)를 포함한다.

요구 사항

설치

FuriosaAI Python SDK는 PyPi를 활용하여 설치할 수 있도록 PyPi 저장소에 준비되어 있다. Python 라이브러리 인터페이스 이외에도 명령 줄 도구 및 다양한 기능을 사용할 수 있고, 다음의 예시대로 설치 가능하다.

# FuriosaAI NPU Python SDK 설치, Python 인터페이스 사용 가능, e.g. `import furiosa`
pip install --upgrade furiosa-sdk
# 부가 도구 설치, 자세한 목록은 아래 참조
pip install --upgrade furiosa-sdk[runtime,quantizer]
# 부가 도구 전체 설치
pip install --upgrade furiosa-sdk[full]

PIP 커맨드를 이용하여 다음 부가 패키지를 설치할 수 있다.

  • cli: 명령 줄 도구 설치, 사용 방법은 명령 줄 도구 빠르게 시작하기 를 참고

    pip install --upgrade furiosa-sdk[cli]
    
  • runtime: FuriosaAI NPU Runtime 을 사용하여 NPU 위에서 모델을 가속시키기 위한 각종 라이브러리 설치, NPU 위에서 모델 가속을 위해 필수

    pip install --upgrade furiosa-sdk[runtime]
    
  • quantizer: 모델의 양자화 도구 설치 (모델 양자화 참고)

    pip install --upgrade furiosa-sdk[quantizer]
    
  • validator: 모델 분석 도구 설치, 해당 모델이 NPU 위에서 가속되기 위해 양자화, 컴파일이 잘 수행되는지 확인하는 도구를 포함

    pip install --upgrade furiosa-sdk[quantizer,runtime,validator,cli]
    

예를 들어 모델 추론을 위한 개발환경과 모델 양자화 도구가 필요한 경우 아래와 같이 설치한다.

pip install --upgrade furiosa-sdk[runtime,quantizer]

Jupyter Notebook 사용 안내

Jupyter Notebook을 사용하는 경우 FuriosaAI Python SDK와 다양한 Python 생태계의 다양한 라이브러리를 편하게 사용할 수 있다.

위 설명에 따라 Python SDK를 이미 설치했다면 아래와 같이 pip를 이용해 Jupyter notebook을 간단히 설치해 사용할 수 있다. Jupyter notebook은 아주 다양한 의존된 패키지를 설치하기 때문에 Conda를 이용한 Python 환경 구성 을 권장한다.

$ pip install jupyterlab
$ jupyter-notebook