Skip to content

GitLab

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

chore: support torch2.8 wheels

parent b17908bc
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
scripts/build_linux_wheel_torch2.8.sh
View file @ 45afb58b
...@@ -22,7 +22,7 @@ docker run --rm \ ...@@ -22,7 +22,7 @@ docker run --rm \
cd /nunchaku && \ cd /nunchaku && \
rm -rf build && \ rm -rf build && \
gcc --version && g++ --version && \ 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 && \ ${PYTHON_ROOT_PATH}/bin/pip install build ninja wheel setuptools && \
export NUNCHAKU_INSTALL_MODE=ALL && \ export NUNCHAKU_INSTALL_MODE=ALL && \
export NUNCHAKU_BUILD_WHEELS=1 && \ export NUNCHAKU_BUILD_WHEELS=1 && \
......
scripts/build_windows_wheel_torch2.8.cmd
View file @ 45afb58b
...@@ -20,7 +20,7 @@ call conda activate %ENV_NAME% ...@@ -20,7 +20,7 @@ call conda activate %ENV_NAME%
:: install dependencies :: install dependencies
call pip install ninja setuptools wheel build 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 environment variables
set NUNCHAKU_INSTALL_MODE=ALL 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