Skip to content
Merged
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
11 changes: 6 additions & 5 deletions .github/workflows/wheels.yml
Original file line number Diff line number Diff line change
Expand Up @@ -198,14 +198,15 @@ jobs:
with:
name: cuda-wheel-${{ matrix.cuda }}
path: cuda
# Both wheels are named as files, so pip takes them whatever their
# version, and the extra's requirement is the CUDA wheel given beside it.
# torch comes from its CUDA index alone: PyPI's torch can be newer and
# built for another CUDA major. Both wheels are named as files, so pip
# takes them whatever their version, and the extra's requirement is the
# CUDA wheel given beside it.
- name: Install blochsim[cu${{ matrix.cuda }}] with torch's ${{ matrix.torch }} build
run: |
python -m pip install --upgrade pip
pip install "$(ls base/*manylinux*x86_64*.whl)[cu${{ matrix.cuda }}]" cuda/*.whl \
torch --index-url https://download.pytorch.org/whl/${{ matrix.torch }} \
--extra-index-url https://pypi.org/simple
pip install torch --index-url https://download.pytorch.org/whl/${{ matrix.torch }}
pip install "$(ls base/*manylinux*x86_64*.whl)[cu${{ matrix.cuda }}]" cuda/*.whl
pip list | grep -Ei '^(torch|blochsim|nvidia-cuda-runtime)'
- name: The card's module loads bare, against torch's runtime
run: python scripts/check_wheel.py
Expand Down
Loading