Installing Dragon

Build
Platform
Language
Acceleration
Distribution

Installation.withOption(stable, linux, windows, macos, Python, cpu, pip)

pip install seeta-dragon

Installation.withOption(stable, linux, windows, python, cu102, pip)

pip install seeta-dragon --no-index -f https://dragon.seetatech.com/download/whl/cu102

Installation.withOption(stable, linux, windows, python, cu110, pip)

pip install seeta-dragon --no-index -f https://dragon.seetatech.com/download/whl/cu110

Installation.withOption(preview, linux, windows, macos, python, cpu, pip)

pip install seeta-dragon --no-index -f https://dragon.seetatech.com/download/whl/nightly/cpu

Installation.withOption(preview, linux, windows, python, cu102, pip)

pip install seeta-dragon --no-index -f https://dragon.seetatech.com/download/whl/nightly/cu102

Installation.withOption(preview, linux, windows, python, cu110, pip)

pip install seeta-dragon --no-index -f https://dragon.seetatech.com/download/whl/nightly/cu110

Get the Source.withOption(preview, stable, linux, python, cu102, cu110, source)

git clone --recursive https://github.com/seetaresearch/dragon.git

We will call the directory that you cloned dragon as SOURCE_DIR.

Installation.withOption(preview, stable, linux, python, cu102, cu110, source)

Step 1: Install C++ dependencies

sudo apt-get install cmake libprotobuf-dev protobuf-compiler

$ Install CUDA

$ Install CUDNN

Step 2: Install Python requirements

pip install -r $SOURCE_DIR/requirements.txt

Step 3: Install Dragon

cd $SOURCE_DIR && mkdir build && cd build
cmake .. -DUSE_CUDA=ON -DUSE_CUDNN=ON
make install -j $(nproc) && pip install ..

Get the Source.withOption(preview, stable, linux, python, cpu, source)

git clone --recursive https://github.com/seetaresearch/dragon.git

We will call the directory that you cloned dragon as SOURCE_DIR.

Installation.withOption(preview, stable, linux, python, cpu, source)

Step 1: Install C++ dependencies

sudo apt-get install cmake libprotobuf-dev protobuf-compiler

Step 2: Install Python requirements

pip install -r $SOURCE_DIR/requirements.txt

Step 3: Install Dragon

cd $SOURCE_DIR && mkdir build && cd build
cmake .. -DUSE_CUDA=OFF -DUSE_CUDNN=OFF
make install -j $(nproc) && pip install ..

Get the Source.withOption(preview, stable, windows, python, cu102, cu110, source)

git clone --recursive https://github.com/seetaresearch/dragon.git

We will call the directory that you cloned dragon as SOURCE_DIR.

Installation.withOption(preview, stable, windows, python, cu102, cu110, source)

Step 1: Install C++ dependencies

$ Install CMake

$ Install ProtoBuf to $SOURCE_DIR/third_party/protobuf

$ Install CUDA

$ Install CUDNN

Step 2: Install Python requirements

pip install -r $SOURCE_DIR/requirements.txt

Step 3: Install Dragon

cd $SOURCE_DIR/scripts
start build_windows_vc[version].bat

Get the Source.withOption(preview, stable, windows, python, cpu, source)

git clone --recursive https://github.com/seetaresearch/dragon.git

We will call the directory that you cloned dragon as SOURCE_DIR.

Installation.withOption(preview, stable, windows, python, cpu, source)

Step 1: Install C++ dependencies

$ Install CMake

$ Install ProtoBuf to $SOURCE_DIR/third_party/protobuf

Step 2: Install Python requirements

pip install -r $SOURCE_DIR/requirements.txt

Step 3: Install Dragon

cd $SOURCE_DIR/scripts
start build_windows_vc[version].bat

Note: Update the batch file to disable CUDA following the linux platform.

Installation.withOption(preview, stable, macos, python, cu102, cu110, source)

Build with CUDA on macOS is deprecated. For acceleration, build with MPS instead.

Get the Source.withOption(preview, stable, macos, python, cpu, source)

git clone --recursive https://github.com/seetaresearch/dragon.git

We will call the directory that you cloned dragon as SOURCE_DIR.

Installation.withOption(preview, stable, macos, python, cpu, source)

Step 1: Install C++ dependencies

brew install cmake protobuf libomp

Step 2: Install Python requirements

pip install -r $SOURCE_DIR/requirements.txt

Step 3: Install Dragon

cd $SOURCE_DIR && mkdir build && cd build
cmake .. \
  -DUSE_CUDA=OFF -DUSE_CUDNN=OFF \
  -DUSE_AVX=OFF -DUSE_AVX2=OFF -DUSE_FMA=OFF \
  -DUSE_MPS=ON -DUSE_BLAS=ON
make install -j $(sysctl -n hw.ncpu) && pip install ..