Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
OpenDAS
torch-scatter
Commits
1af79dd6
"...git@developer.sourcefind.cn:renzhc/diffusers_dcu.git" did not exist on "ac4c695d97779cc4d0c92585f8e87ad39aa4aeb2"
Commit
1af79dd6
authored
Jun 16, 2021
by
rusty1s
Browse files
[skip ci] update cuda installation
parent
19937735
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
23 additions
and
3 deletions
+23
-3
.github/workflows/building.yml
.github/workflows/building.yml
+23
-3
No files found.
.github/workflows/building.yml
View file @
1af79dd6
...
@@ -46,10 +46,30 @@ jobs:
...
@@ -46,10 +46,30 @@ jobs:
run
:
|
run
:
|
pip install torch==${{ matrix.torch-version}}+${{ matrix.cuda-version }} -f https://download.pytorch.org/whl/torch_stable.html
pip install torch==${{ matrix.torch-version}}+${{ matrix.cuda-version }} -f https://download.pytorch.org/whl/torch_stable.html
-
name
:
Build main package
-
Install main package for CPU
:
if
:
${{ matrix.cuda-version != 'cpu' }}
run
:
|
run
:
|
pip install wheel
source .github/workflows/cuda/${{ matrix.cuda-version }}-${{ runner.os }}-env.sh
source .github/workflows/cuda/${{ matrix.cuda-version }}-${{ runner.os }}-env.sh
python setup.py bdist_wheel --dist-dir=dist
FORCE_CUDA=1 pip install -e .
ls -lah torch_scatter/
-
Install main package with CUDA
:
if
:
${{ matrix.cuda-version == 'cpu' }}
run
:
|
FORCE_ONLY_CPU=1 pip install -e .
ls -lah torch_scatter/
ls -lah torch_scatter/
-
name
:
Build wheel
run
:
|
pip install wheel
python setup.py bdist_wheel --dist-dir=dist
ls -lah 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
}}'
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment