Commit abb2e1dc authored by rusty1s's avatar rusty1s
Browse files

fix

parent 327b8b1a
......@@ -48,7 +48,7 @@ jobs:
- CUDA=10.1.105-1
- UBUNTU_VERSION=ubuntu1804
- CUBLAS=libcublas-dev
- TORCH=${TORCH_VERSION}+
- TORCH=${TORCH_VERSION}
- name: macosx/cpu
os: osx
osx_image: xcode11.3
......@@ -98,7 +98,7 @@ jobs:
# - choco install python --version 3.7.6
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
- pip3 install numpy
- pip3 install torch==${TORCH} -f https://download.pytorch.org/whl/torch_stable.html
......
......@@ -15,7 +15,8 @@ from .composite import (scatter_std, scatter_logsumexp, scatter_softmax,
torch.ops.load_library(
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'
......
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