Skip to content

GitLab

Commit abb2e1dc authored by rusty1s's avatar rusty1s
Browse files

fix

parent 327b8b1a
Hide whitespace changes
Inline Side-by-side
Showing with 4 additions and 3 deletions
.travis.yml
View file @ abb2e1dc
...@@ -48,7 +48,7 @@ jobs: ...@@ -48,7 +48,7 @@ jobs:
- CUDA=10.1.105-1 - CUDA=10.1.105-1
- UBUNTU_VERSION=ubuntu1804 - UBUNTU_VERSION=ubuntu1804
- CUBLAS=libcublas-dev - CUBLAS=libcublas-dev
- TORCH=${TORCH_VERSION}+ - TORCH=${TORCH_VERSION}
- name: macosx/cpu - name: macosx/cpu
os: osx os: osx
osx_image: xcode11.3 osx_image: xcode11.3
...@@ -98,7 +98,7 @@ jobs: ...@@ -98,7 +98,7 @@ jobs:
# - choco install python --version 3.7.6 # - choco install python --version 3.7.6
install: install:
- if [ "${TRAVIS_OS_NAME}" = "windows" ]; then alias python3=python; fi - if [ "${TRAVIS_OS_NAME}" = "windows" ]; then echo "wadawd"; alias python3=python; fi
- source install.sh - source install.sh
- pip3 install numpy - pip3 install numpy
- pip3 install torch==${TORCH} -f https://download.pytorch.org/whl/torch_stable.html - pip3 install torch==${TORCH} -f https://download.pytorch.org/whl/torch_stable.html
......
torch_scatter/__init__.py
View file @ abb2e1dc
...@@ -15,7 +15,8 @@ from .composite import (scatter_std, scatter_logsumexp, scatter_softmax, ...@@ -15,7 +15,8 @@ from .composite import (scatter_std, scatter_logsumexp, scatter_softmax,
torch.ops.load_library( torch.ops.load_library(
osp.join(osp.dirname(osp.abspath(__file__)), '_version.so')) osp.join(osp.dirname(osp.abspath(__file__)), '_version.so'))
_version = torch.ops.torch_scatter.cuda_version() cuda_version = torch.ops.torch_scatter.cuda_version()
print(cuda_version)
__version__ = '2.0.3' __version__ = '2.0.3'
......
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