콘텐츠로 이동

설치

torch-rbln은 사전 빌드된 휠로 설치하거나 소스에서 빌드할 수 있습니다. RBLN NPU용으로 모델을 컴파일하려면 rebel-compiler를 별도로 설치해야 합니다.

Python에서는 import torch를 사용하면 됩니다. torch-rbln은 out-of-tree extension으로 자동 로드됩니다.

1
import torch

버전 이력은 릴리즈 노트를 참조하세요.

빠른 시작

가상 환경을 생성한 뒤 필요한 패키지를 설치하고, Python에서 모듈 가져오기(import)가 정상인지 확인합니다.

1
2
3
python3 -m venv .venv
source .venv/bin/activate
python -m pip install -U pip

1
2
3
$ pip install \
  --extra-index-url https://download.pytorch.org/whl/cpu \
  torch-rbln==0.1.8

1
2
3
$ pip install \
  --extra-index-url https://pypi.rbln.ai/simple \
  rebel-compiler==0.10.2

1
python -c "import torch; import torch_rbln; print(torch_rbln.__version__)"

소스 빌드 (고급)

저장소: torch-rbln

요구 사항

  • Python: 3.10–3.13
  • PyTorch (CPU 휠)
  • rebel-compiler (필수, 별도 설치)

Note

rebel-compiler 설치에는 RBLN Portal 계정이 필요합니다.

전제 조건

  • Git
  • torch-rbln 확장 빌드를 위한 C/C++ 툴체인(Linux에서는 GCC 13)
  • CMake 3.18+ 및 Ninja
  • uv (선택). uv 설치 가이드를 참조하세요.

소스 받기

저장소를 클론하고 가상 환경을 만듭니다.

1
2
3
4
git clone https://github.com/RBLN-SW/torch-rbln.git
cd torch-rbln
uv venv .venv && source .venv/bin/activate
uv pip install -U pip

dev-setup.sh 사용 (권장)

저장소 루트에서 설정 스크립트를 실행합니다.

1
./tools/dev-setup.sh pypi

설치 확인은 빠른 시작의 명령을 사용합니다.

수동 빌드

저장소 루트에서:

1
2
3
uv sync --no-install-project
# export TORCH_RBLN_BUILD_TYPE=Debug   # 선택
uv pip install -e . --no-build-isolation

설치 확인은 빠른 시작의 명령을 사용합니다.