Skip to content

installation.md

Latest commit

 

History

History
58 lines (39 loc) · 1.25 KB

installation.md

File metadata and controls

58 lines (39 loc) · 1.25 KB

Installation

As editable

First, clone the Conch repo.

git clone https://github.com/stackav-oss/conch.git
cd conch

then install:

pip install -e .

Torch/Triton installation

To install torch/triton, specify either the cuda or rocm extra (depending on your platform).

# For Nvidia/CUDA
pip install -e ".[cuda]"
# For AMD/ROCm
pip install -e ".[rocm]" --extra-index-url https://download.pytorch.org/whl/rocm6.2.4

As wheel

You can install conch from PyPi via:

pip install conch-triton-kernels

Note: by default, without any extras specified, this will not install torch or triton. This allows usage of Conch as long as Torch and Triton are already installed in your environment.

Nvidia/CUDA

For Nvidia/CUDA platforms, you can specify the [cuda] extra to install torch and triton for Nvidia/CUDA platforms.

pip install "conch-triton-kernels[cuda]"

AMD/ROCm

For AMD/ROCm platforms, you can specify the [rocm] extra to install torch and triton for AMD/ROCm platforms. You must also specify the appropriate --extra-index-url.

pip install "conch-triton-kernels[rocm]" --extra-index-url https://download.pytorch.org/whl/rocm6.2.4