Commit c6000fcb authored by rusty1s's avatar rusty1s
Browse files

[skip ci] update cuda installation

parent 111f2270
...@@ -10,9 +10,9 @@ jobs: ...@@ -10,9 +10,9 @@ jobs:
strategy: strategy:
matrix: matrix:
os: [ubuntu-18.04] os: [ubuntu-18.04]
python-version: [3.9] python-version: [3.6, 3.7, 3.8, 3.9]
torch-version: [1.9.0] torch-version: [1.9.0]
cuda-version: ['cu102'] cuda-version: ['cpu', 'cu102', 'cu111']
steps: steps:
- uses: actions/checkout@v2 - uses: actions/checkout@v2
...@@ -53,11 +53,11 @@ jobs: ...@@ -53,11 +53,11 @@ jobs:
python setup.py bdist_wheel --dist-dir=dist python setup.py bdist_wheel --dist-dir=dist
ls -lah dist/ ls -lah dist/
# - name: Upload wheel - name: Upload wheel
# uses: shallwefootball/s3-upload-action@master uses: shallwefootball/s3-upload-action@master
# with: with:
# aws_key_id: ${{ secrets.AWS_KEY_ID }} aws_key_id: ${{ secrets.AWS_KEY_ID }}
# aws_secret_access_key: ${{ secrets.AWS_SECRET_ACCESS_KEY}} aws_secret_access_key: ${{ secrets.AWS_SECRET_ACCESS_KEY}}
# aws_bucket: 'pytorch-geometric.com' aws_bucket: 'pytorch-geometric.com'
# source_dir: 'dist' source_dir: 'dist'
# destination_dir: 'whl/torch-${{ matrix.torch-version }}+${{ matrix.cuda-version }}' destination_dir: 'whl/torch-${{ matrix.torch-version }}+${{ matrix.cuda-version }}'
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment