Commit 4f946374 authored by rusty1s's avatar rusty1s
Browse files

update

parent e09a12b4
...@@ -49,15 +49,13 @@ jobs: ...@@ -49,15 +49,13 @@ jobs:
- name: Install PyTorch ${{ matrix.torch-version }}+${{ matrix.cuda-version }} - name: Install PyTorch ${{ matrix.torch-version }}+${{ matrix.cuda-version }}
run: | run: |
pip install numpy typing-extensions dataclasses pip install --no-index --no-cache-dir torch==${{ matrix.torch-version }} -f https://download.pytorch.org/whl/${{ matrix.cuda-version }}
pip install --no-index --no-cache-dir torch==${{ matrix.torch-version}} -f https://download.pytorch.org/whl/${{ matrix.cuda-version }}/torch_stable.html
python -c "import torch; print('PyTorch:', torch.__version__)" python -c "import torch; print('PyTorch:', torch.__version__)"
python -c "import torch; print('CUDA:', torch.version.cuda)" python -c "import torch; print('CUDA:', torch.version.cuda)"
- name: Set version - name: Set version
run: | run: |
VERSION=`sed -n "s/^__version__ = '\(.*\)'/\1/p" torch_spline_conv/__init__.py` VERSION=`sed -n "s/^__version__ = '\(.*\)'/\1/p" torch_spline_conv/__init__.py`
sed -i "s/$VERSION/$VERSION+{{ matrix.cuda-version }}/" setup.py
sed -i "s/$VERSION/$VERSION+{{ matrix.cuda-version }}/" torch_spline_conv/__init__.py sed -i "s/$VERSION/$VERSION+{{ matrix.cuda-version }}/" torch_spline_conv/__init__.py
- name: Install main package for CPU - name: Install main package for CPU
......
...@@ -44,7 +44,7 @@ jobs: ...@@ -44,7 +44,7 @@ jobs:
- name: Install PyTorch ${{ matrix.torch-version }} - name: Install PyTorch ${{ matrix.torch-version }}
run: | run: |
pip install torch==${{ matrix.torch-version}}+cpu -f https://download.pytorch.org/whl/torch_stable.html pip install --no-index --no-cache-dir torch==${{ matrix.torch-version }} -f https://download.pytorch.org/whl/cpu
- name: Install dependencies - name: Install dependencies
run: | run: |
......
...@@ -27,7 +27,7 @@ jobs: ...@@ -27,7 +27,7 @@ jobs:
- name: Install PyTorch ${{ matrix.torch-version }} - name: Install PyTorch ${{ matrix.torch-version }}
run: | run: |
pip install torch==${{ matrix.torch-version}}+cpu -f https://download.pytorch.org/whl/torch_stable.html pip install --no-index --no-cache-dir torch==${{ matrix.torch-version }} -f https://download.pytorch.org/whl/cpu
- name: Install main package - name: Install main package
run: | run: |
......
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