Skip to content

[pre-commit.ci] pre-commit autoupdate #2111

[pre-commit.ci] pre-commit autoupdate

[pre-commit.ci] pre-commit autoupdate #2111

Workflow file for this run

name: Minimal Installation
on: # yamllint disable-line rule:truthy
push:
branches:
- master
pull_request:
jobs:
import:
runs-on: ubuntu-latest
strategy:
matrix:
cuda-version: ['cpu', 'cu121']
steps:
- name: Checkout repository
uses: actions/checkout@v4
with:
submodules: recursive
- name: Setup packages
uses: ./.github/actions/setup
with:
cuda-version: ${{ matrix.cuda-version }}
- name: Install package
run: |
source ./.github/workflows/cuda/${{ runner.os }}-env.sh ${{ matrix.cuda-version }}
pip install --verbose -e .
shell: bash
env:
TORCH_CUDA_ARCH_LIST: "5.0+PTX;6.0;7.0;7.5;8.0;8.6"
- name: Test imports
run: |
python -c 'import pyg_lib;print(pyg_lib.cuda_version())'
python -c "import pyg_lib.ops"
python -c "import pyg_lib.sampler"