Commit c6000fcb authored by rusty1s's avatar rusty1s
Browse files

[skip ci] update cuda installation

parent 111f2270
......@@ -10,9 +10,9 @@ jobs:
strategy:
matrix:
os: [ubuntu-18.04]
python-version: [3.9]
python-version: [3.6, 3.7, 3.8, 3.9]
torch-version: [1.9.0]
cuda-version: ['cu102']
cuda-version: ['cpu', 'cu102', 'cu111']
steps:
- uses: actions/checkout@v2
......@@ -53,11 +53,11 @@ jobs:
python setup.py bdist_wheel --dist-dir=dist
ls -lah dist/
# - name: Upload wheel
# uses: shallwefootball/s3-upload-action@master
# with:
# aws_key_id: ${{ secrets.AWS_KEY_ID }}
# aws_secret_access_key: ${{ secrets.AWS_SECRET_ACCESS_KEY}}
# aws_bucket: 'pytorch-geometric.com'
# source_dir: 'dist'
# destination_dir: 'whl/torch-${{ matrix.torch-version }}+${{ matrix.cuda-version }}'
- name: Upload wheel
uses: shallwefootball/s3-upload-action@master
with:
aws_key_id: ${{ secrets.AWS_KEY_ID }}
aws_secret_access_key: ${{ secrets.AWS_SECRET_ACCESS_KEY}}
aws_bucket: 'pytorch-geometric.com'
source_dir: 'dist'
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