Skip to content

Commit

Permalink
fix wheels_build run failed
Browse files Browse the repository at this point in the history
  • Loading branch information
chengzeyi committed Nov 12, 2023
1 parent 22ad27e commit bf3853b
Show file tree
Hide file tree
Showing 2 changed files with 15 additions and 19 deletions.
32 changes: 14 additions & 18 deletions .github/workflows/wheels_build.yml
Original file line number Diff line number Diff line change
Expand Up @@ -90,6 +90,13 @@ jobs:
sh ./cuda.run --silent --toolkit &&
rm ./cuda.run
- name: Recursive checkout
uses: actions/checkout@v3
with:
submodules: recursive
path: "."
fetch-depth: 0 # for tags

- if: runner.os != 'Windows'
name: (Linux) Setup venv for linux
run: |
Expand All @@ -100,24 +107,6 @@ jobs:
echo "PATH=$PATH" >> ${GITHUB_ENV}
echo "MAX_JOBS=3" >> ${GITHUB_ENV}
# Pre-install some to generate cache
- name: Install dependencies
run: $PY -m pip install wheel setuptools twine "torch==${{ inputs.torch_version }}" --extra-index-url https://download.pytorch.org/whl/cu${{ inputs.cuda_short_version }} --no-cache-dir

- name: Recursive checkout
uses: actions/checkout@v3
with:
submodules: recursive
path: "."
fetch-depth: 0 # for tags

- if: runner.os == 'Windows'
name: (Windows) Setup Runner
uses: ./.github/actions/setup-windows-runner
with:
cuda: ${{ steps.cuda_info.outputs.CUDA_VERSION }}
python: ${{ inputs.python }}

- name: Define version
id: sfast_version
env:
Expand Down Expand Up @@ -145,6 +134,13 @@ jobs:
echo "torch == ${{ inputs.torch_version }}" >> ./requirements.txt
cat ./requirements.txt
- if: runner.os == 'Windows'
name: (Windows) Setup Runner
uses: ./.github/actions/setup-windows-runner
with:
cuda: ${{ steps.cuda_info.outputs.CUDA_VERSION }}
python: ${{ inputs.python }}

- name: Install dependencies
run: $PY -m pip install wheel setuptools twine -r requirements.txt --extra-index-url https://download.pytorch.org/whl/cu${{ inputs.cuda_short_version }} --no-cache-dir

Expand Down
2 changes: 1 addition & 1 deletion version.txt
Original file line number Diff line number Diff line change
@@ -1 +1 @@
0.0.8
0.0.9

0 comments on commit bf3853b

Please sign in to comment.