Skip to content

Commit 34c7127

Browse files
lmxyylmxyy
committed
done
1 parent 3deb7cf commit 34c7127

6 files changed

Lines changed: 21 additions & 14 deletions

File tree

‎.github/workflows/nightly-build.yaml‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -135,7 +135,7 @@ jobs:
135135
- name: Build wheels
136136
shell: cmd
137137
run: |
138-
call C:\Users\muyangl\miniconda3\condabin\activate.bat activate
138+
call C:\Users\nunchaku\miniconda3\condabin\activate.bat activate
139139
IF "${{ matrix.torch }}"=="pre" (
140140
call scripts\build_windows_wheel_torch_nightly.cmd ${{ matrix.python }} ${{ matrix.torch }} ${{ matrix.cuda }}
141141
) ELSE (

‎.github/workflows/release-build.yaml‎

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -120,7 +120,7 @@ jobs:
120120
- name: Build wheels
121121
shell: cmd
122122
run: |
123-
call C:\Users\muyangl\miniconda3\condabin\activate.bat activate
123+
call C:\Users\nunchaku\miniconda3\condabin\activate.bat activate
124124
IF "${{ matrix.torch }}"=="pre" (
125125
call scripts\build_windows_wheel_torch_nightly.cmd ${{ matrix.python }} ${{ matrix.torch }} ${{ matrix.cuda }}
126126
) ELSE (
@@ -140,7 +140,7 @@ jobs:
140140
# env:
141141
# HF_TOKEN: ${{ secrets.HF_WHEEL_UPLOAD_TOKEN }}
142142
# run: |
143-
# call C:\Users\muyangl\miniconda3\condabin\activate.bat activate
143+
# call C:\Users\nunchaku\miniconda3\condabin\activate.bat activate
144144
# pip install huggingface_hub
145145
# for %%w in (dist\*.whl) do (
146146
# hf upload nunchaku-tech/nunchaku "%%w"

‎scripts/build_linux_wheel.sh‎

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -30,8 +30,9 @@ docker run --rm \
3030
cd /nunchaku && \
3131
rm -rf build && \
3232
gcc --version && g++ --version && \
33-
${PYTHON_ROOT_PATH}/bin/pip install --no-cache-dir torch==${TORCH_VERSION} torchvision==${TORCHVISION_VERSION} torchaudio==${TORCHAUDIO_VERSION} --index-url https://download.pytorch.org/whl/cu${CUDA_VERSION//.} && \
34-
${PYTHON_ROOT_PATH}/bin/pip install build ninja wheel setuptools && \
33+
${PYTHON_ROOT_PATH}/bin/pip install uv && \
34+
${PYTHON_ROOT_PATH}/bin/uv pip install --no-cache-dir torch==${TORCH_VERSION} torchvision==${TORCHVISION_VERSION} torchaudio==${TORCHAUDIO_VERSION} --index-url https://download.pytorch.org/whl/cu${CUDA_VERSION//.} && \
35+
${PYTHON_ROOT_PATH}/bin/uv pip install build ninja wheel setuptools && \
3536
export NUNCHAKU_INSTALL_MODE=ALL && \
3637
export NUNCHAKU_BUILD_WHEELS=1 && \
3738
export MAX_JOBS=${MAX_JOBS} && \

‎scripts/build_linux_wheel_torch_nightly.sh‎

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -22,8 +22,9 @@ docker run --rm \
2222
cd /nunchaku && \
2323
rm -rf build && \
2424
gcc --version && g++ --version && \
25-
${PYTHON_ROOT_PATH}/bin/pip install --pre --no-cache-dir torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu${CUDA_VERSION//.} && \
26-
${PYTHON_ROOT_PATH}/bin/pip install build ninja wheel setuptools && \
25+
${PYTHON_ROOT_PATH}/bin/pip install uv && \
26+
${PYTHON_ROOT_PATH}/bin/uv pip install --pre --no-cache-dir torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu${CUDA_VERSION//.} && \
27+
${PYTHON_ROOT_PATH}/bin/uv pip install build ninja wheel setuptools && \
2728
export NUNCHAKU_INSTALL_MODE=ALL && \
2829
export NUNCHAKU_BUILD_WHEELS=1 && \
2930
export MAX_JOBS=${MAX_JOBS} && \

‎scripts/build_windows_wheel.cmd‎

Lines changed: 6 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -34,19 +34,22 @@ call conda create -y -n %ENV_NAME% python=%PYTHON_VERSION%
3434
call conda activate %ENV_NAME%
3535

3636
:: install dependencies
37-
call pip install ninja setuptools wheel build
38-
call pip install --no-cache-dir torch==%TORCH_VERSION% torchvision==%TORCHVISION_VERSION% torchaudio==%TORCHAUDIO_VERSION% --index-url "https://download.pytorch.org/whl/cu%CUDA_SHORT_VERSION%/"
37+
call pip install uv
38+
call uv pip install ninja setuptools wheel build
39+
call uv pip install --no-cache-dir torch==%TORCH_VERSION% torchvision==%TORCHVISION_VERSION% torchaudio==%TORCHAUDIO_VERSION% --index-url "https://download.pytorch.org/whl/cu%CUDA_SHORT_VERSION%/"
3940

4041
:: set environment variables
4142
set NUNCHAKU_INSTALL_MODE=ALL
4243
set NUNCHAKU_BUILD_WHEELS=1
44+
set NVCC_PREPEND_FLAGS=-allow-unsupported-compiler
45+
set CUDA_HOME=C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v%CUDA_VERSION%
4346

4447
:: cd to the parent directory
4548
cd /d "%~dp0.."
4649
if exist build rd /s /q build
4750

4851
:: set up Visual Studio compilation environment
49-
call "C:\Program Files (x86)\Microsoft Visual Studio\2022\BuildTools\Common7\Tools\VsDevCmd.bat" -startdir=none -arch=x64 -host_arch=x64
52+
call "C:\Program Files (x86)\Microsoft Visual Studio\18\BuildTools\Common7\Tools\VsDevCmd.bat" -startdir=none -arch=x64 -host_arch=x64
5053
set DISTUTILS_USE_SDK=1
5154

5255
:: build wheels

‎scripts/build_windows_wheel_torch_nightly.cmd‎

Lines changed: 6 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -18,20 +18,22 @@ call conda create -y -n %ENV_NAME% python=%PYTHON_VERSION%
1818
call conda activate %ENV_NAME%
1919

2020
:: install dependencies
21-
call pip install ninja setuptools wheel build
22-
23-
call pip install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu%CUDA_SHORT_VERSION%
21+
call pip install uv
22+
call uv pip install ninja setuptools wheel build
23+
call uv pip install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu%CUDA_SHORT_VERSION%
2424

2525
:: set environment variables
2626
set NUNCHAKU_INSTALL_MODE=ALL
2727
set NUNCHAKU_BUILD_WHEELS=1
28+
set NVCC_PREPEND_FLAGS=-allow-unsupported-compiler
29+
set CUDA_HOME=C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v%CUDA_VERSION%
2830

2931
:: cd to the parent directory
3032
cd /d "%~dp0.."
3133
if exist build rd /s /q build
3234

3335
:: set up Visual Studio compilation environment
34-
call "C:\Program Files (x86)\Microsoft Visual Studio\2022\BuildTools\Common7\Tools\VsDevCmd.bat" -startdir=none -arch=x64 -host_arch=x64
36+
call "C:\Program Files (x86)\Microsoft Visual Studio\18\BuildTools\Common7\Tools\VsDevCmd.bat" -startdir=none -arch=x64 -host_arch=x64
3537
set DISTUTILS_USE_SDK=1
3638

3739
:: build wheels

0 commit comments

Comments
 (0)