Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
fengzch-das
nunchaku
Commits
45afb58b
Commit
45afb58b
authored
Jun 03, 2025
by
muyangli
Browse files
chore: support torch2.8 wheels
parent
b17908bc
Changes
2
Hide whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
2 additions
and
2 deletions
+2
-2
scripts/build_linux_wheel_torch2.8.sh
scripts/build_linux_wheel_torch2.8.sh
+1
-1
scripts/build_windows_wheel_torch2.8.cmd
scripts/build_windows_wheel_torch2.8.cmd
+1
-1
No files found.
scripts/build_linux_wheel_torch2.8.sh
View file @
45afb58b
...
...
@@ -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 &&
\
...
...
scripts/build_windows_wheel_torch2.8.cmd
View file @
45afb58b
...
...
@@ -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
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment