Commit 45afb58b authored by muyangli's avatar muyangli
Browse files

chore: support torch2.8 wheels

parent b17908bc
......@@ -22,7 +22,7 @@ docker run --rm \
cd /nunchaku && \
rm -rf build && \
gcc --version && g++ --version && \
${PYTHON_ROOT_PATH}/bin/pip install --pre --no-cache-dir torch==${TORCH_VERSION} torchvision==${TORCHVISION_VERSION} torchaudio==${TORCHAUDIO_VERSION} --index-url https://download.pytorch.org/whl/nightly/cu${CUDA_VERSION//.} && \
${PYTHON_ROOT_PATH}/bin/pip install --pre --no-cache-dir torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu${CUDA_VERSION//.} && \
${PYTHON_ROOT_PATH}/bin/pip install build ninja wheel setuptools && \
export NUNCHAKU_INSTALL_MODE=ALL && \
export NUNCHAKU_BUILD_WHEELS=1 && \
......
......@@ -20,7 +20,7 @@ call conda activate %ENV_NAME%
:: install dependencies
call pip install ninja setuptools wheel build
call pip install --pre torch==2.8 torchvision==0.23 torchaudio==2.8 --index-url https://download.pytorch.org/whl/nightly/cu128
call pip install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu128
:: set environment variables
set NUNCHAKU_INSTALL_MODE=ALL
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment