Error running in a new installation!

[abose17]

I have installed pytorch 1.10.1 and follwoing is the information about my Cuda compilation tools, release 10.1, V10.1.243. I installed torch_sparce and torch_scatter according to the torch and cuda version. Still getting this issue

Traceback (most recent call last):
  File "/home/abose/anaconda3/envs/pyTorch_env/lib/python3.8/runpy.py", line 194, in _run_module_as_main
    return _run_code(code, main_globals, None,
  File "/home/abose/anaconda3/envs/pyTorch_env/lib/python3.8/runpy.py", line 87, in _run_code
    exec(code, run_globals)
  File "/home/abose/Graph Convolution Core/pytorch_geo_Monet/graph/test.py", line 2, in <module>
    from torch_geometric.datasets import Planetoid
  File "/home/abose/anaconda3/envs/pyTorch_env/lib/python3.8/site-packages/torch_geometric/__init__.py", line 4, in <module>
    import torch_geometric.data
  File "/home/abose/anaconda3/envs/pyTorch_env/lib/python3.8/site-packages/torch_geometric/data/__init__.py", line 1, in <module>
    from .data import Data
  File "/home/abose/anaconda3/envs/pyTorch_env/lib/python3.8/site-packages/torch_geometric/data/data.py", line 9, in <module>
    from torch_sparse import SparseTensor
  File "/home/abose/anaconda3/envs/pyTorch_env/lib/python3.8/site-packages/torch_sparse/__init__.py", line 40, in <module>
    from .storage import SparseStorage  # noqa
  File "/home/abose/anaconda3/envs/pyTorch_env/lib/python3.8/site-packages/torch_sparse/storage.py", line 6, in <module>
    from torch_sparse.utils import Final

ImportError: cannot import name 'Final' from 'torch_sparse.utils' (/home/abose/anaconda3/envs/pyTorch_env/lib/python3.8/site-packages/torch_sparse/utils/init.py)

[rusty1s]

Can you try to run

pip install typing-extensions

[abose17]

Thanks Matthias for your reply.
The following command you mentioned did not work.
$pip install typing-extension

ERROR: Could not find a version that satisfies the requirement typing-extension
ERROR: No matching distribution found for typing-extension

[rusty1s]

Note that you need to append a s :)

pip install typing-extensions

[abose17]

I am sorry, it's my bad (missed to put 's') but still it did not work and showing the same problem again. I went through that path directory and the init.py seems empty.

Traceback (most recent call last):
  File "/home/abose/anaconda3/envs/pyTorch_env/lib/python3.8/runpy.py", line 194, in _run_module_as_main
    return _run_code(code, main_globals, None,
  File "/home/abose/anaconda3/envs/pyTorch_env/lib/python3.8/runpy.py", line 87, in _run_code
    exec(code, run_globals)
  File "/home/abose/Graph Convolution Core/pytorch_geo_Monet/graph/test.py", line 2, in <module>
    from torch_geometric.datasets import Planetoid
  File "/home/abose/anaconda3/envs/pyTorch_env/lib/python3.8/site-packages/torch_geometric/__init__.py", line 4, in <module>
    import torch_geometric.data
  File "/home/abose/anaconda3/envs/pyTorch_env/lib/python3.8/site-packages/torch_geometric/data/__init__.py", line 1, in <module>
    from .data import Data
  File "/home/abose/anaconda3/envs/pyTorch_env/lib/python3.8/site-packages/torch_geometric/data/data.py", line 9, in <module>
    from torch_sparse import SparseTensor
  File "/home/abose/anaconda3/envs/pyTorch_env/lib/python3.8/site-packages/torch_sparse/__init__.py", line 40, in <module>
    from .storage import SparseStorage  # noqa
  File "/home/abose/anaconda3/envs/pyTorch_env/lib/python3.8/site-packages/torch_sparse/storage.py", line 6, in <module>
    from torch_sparse.utils import Final
ImportError: cannot import name 'Final' from 'torch_sparse.utils' (/home/abose/anaconda3/envs/pyTorch_env/lib/python3.8/site-packages/torch_sparse/utils/__init__.py)

[rusty1s]

What happens if you run

from typing_extensions import Final
from torch.jit import Final

[abose17]

Matthias,
It runs okay in terminal python script but it does not help to run the .py files located in my local machine. Do you suggest me to put these lines in any specific .py files?

[rusty1s]

What's the error you get when running this in *.py files? If there is a difference in Terminal and *.py files, I think you have mixed up Python versions. You can also try to install via python -m pip install typing-extensions.

[abose17]

I ran the above script in the terminal and it showed
Requirement already satisfied: typing-extensions in /home/abose/anaconda3/envs/pyTorch_env/lib/python3.8/site-packages (4.1.1)
The init.py file is still empty. I did not face any problem writing the following two lines of code in a *.py file

from typing_extensions import Final
from torch.jit import Final

*Written code inside the .py file

import os.path as osp

from torch_geometric.datasets import Planetoid
from torch_geometric.loader import NeighborLoader
data = Planetoid(osp.dirname(osp.realpath(__file__)), name='Cora')[0]
#print(" data.train_mask size-> ", data.train_mask.size(), " data.train_mask -> ", data.train_mask)

loader = NeighborLoader(
    data,
    # Sample 30 neighbors for each node for 2 iterations
    num_neighbors=[30] * 2,
    # Use a batch size of 128 for sampling training nodes
    batch_size=128,
    input_nodes=data.train_mask,
)

sampled_data = next(iter(loader))
print(sampled_data.batch_size)

Error Report

Traceback (most recent call last):
  File "/home/abose/anaconda3/envs/pyTorch_env/lib/python3.8/runpy.py", line 194, in _run_module_as_main
    return _run_code(code, main_globals, None,
  File "/home/abose/anaconda3/envs/pyTorch_env/lib/python3.8/runpy.py", line 87, in _run_code
    exec(code, run_globals)
  File "/home/abose/Graph Convolution Core/pytorch_geo_Monet/graph/test.py", line 4, in <module>
    from torch_geometric.datasets import Planetoid
  File "/home/abose/anaconda3/envs/pyTorch_env/lib/python3.8/site-packages/torch_geometric/__init__.py", line 4, in <module>
    import torch_geometric.data
  File "/home/abose/anaconda3/envs/pyTorch_env/lib/python3.8/site-packages/torch_geometric/data/__init__.py", line 1, in <module>
    from .data import Data
  File "/home/abose/anaconda3/envs/pyTorch_env/lib/python3.8/site-packages/torch_geometric/data/data.py", line 9, in <module>
    from torch_sparse import SparseTensor
  File "/home/abose/anaconda3/envs/pyTorch_env/lib/python3.8/site-packages/torch_sparse/__init__.py", line 40, in <module>
    from .storage import SparseStorage  # noqa
  File "/home/abose/anaconda3/envs/pyTorch_env/lib/python3.8/site-packages/torch_sparse/storage.py", line 6, in <module>
    from torch_sparse.utils import Final
ImportError: cannot import name 'Final' from 'torch_sparse.utils' (/home/abose/anaconda3/envs/pyTorch_env/lib/python3.8/site-packages/torch_sparse/utils/__init__.py)

[abose17]

I created another conda environment and tried to install a fresh PyG. My machine configuration

My Torch version

>>> torch.__version__
'1.10.1'
nvcc --version
nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2019 NVIDIA Corporation
Built on Sun_Jul_28_19:07:16_PDT_2019
Cuda compilation tools, release 10.1, V10.1.243

While installing I got the following:

pip install torch-scatter -f https://data.pyg.org/whl/torch-1.10.0+${CUDA}.html
Looking in links: https://data.pyg.org/whl/torch-1.10.0+.html
Collecting torch-scatter
  Downloading torch_scatter-2.0.9.tar.gz (21 kB)
Building wheels for collected packages: torch-scatter
  Building wheel for torch-scatter (setup.py) ... error
  ERROR: Command errored out with exit status 1:
   command: /home/abose/anaconda3/envs/pyTorch_Geo/bin/python -u -c 'import io, os, sys, setuptools, tokenize; sys.argv[0] = '"'"'/tmp/pip-install-k2o__zde/torch-scatter_fd431bdb352b4ae3bb0cc51fb985c3d9/setup.py'"'"'; __file__='"'"'/tmp/pip-install-k2o__zde/torch-scatter_fd431bdb352b4ae3bb0cc51fb985c3d9/setup.py'"'"';f = getattr(tokenize, '"'"'open'"'"', open)(__file__) if os.path.exists(__file__) else io.StringIO('"'"'from setuptools import setup; setup()'"'"');code = f.read().replace('"'"'\r\n'"'"', '"'"'\n'"'"');f.close();exec(compile(code, __file__, '"'"'exec'"'"'))' bdist_wheel -d /tmp/pip-wheel-meufphar
       cwd: /tmp/pip-install-k2o__zde/torch-scatter_fd431bdb352b4ae3bb0cc51fb985c3d9/
  Complete output (39 lines):
  /home/abose/anaconda3/envs/pyTorch_Geo/lib/python3.9/site-packages/setuptools/dist.py:757: UserWarning: Usage of dash-separated 'description-file' will not be supported in future versions. Please use the underscore name 'description_file' instead
    warnings.warn(
  running bdist_wheel
  running build
  running build_py
  creating build
  creating build/lib.linux-x86_64-3.9
  creating build/lib.linux-x86_64-3.9/torch_scatter
  copying torch_scatter/segment_coo.py -> build/lib.linux-x86_64-3.9/torch_scatter
  copying torch_scatter/placeholder.py -> build/lib.linux-x86_64-3.9/torch_scatter
  copying torch_scatter/scatter.py -> build/lib.linux-x86_64-3.9/torch_scatter
  copying torch_scatter/segment_csr.py -> build/lib.linux-x86_64-3.9/torch_scatter
  copying torch_scatter/__init__.py -> build/lib.linux-x86_64-3.9/torch_scatter
  copying torch_scatter/utils.py -> build/lib.linux-x86_64-3.9/torch_scatter
  creating build/lib.linux-x86_64-3.9/torch_scatter/composite
  copying torch_scatter/composite/std.py -> build/lib.linux-x86_64-3.9/torch_scatter/composite
  copying torch_scatter/composite/__init__.py -> build/lib.linux-x86_64-3.9/torch_scatter/composite
  copying torch_scatter/composite/softmax.py -> build/lib.linux-x86_64-3.9/torch_scatter/composite
  copying torch_scatter/composite/logsumexp.py -> build/lib.linux-x86_64-3.9/torch_scatter/composite
  running build_ext
  /home/abose/anaconda3/envs/pyTorch_Geo/lib/python3.9/site-packages/torch/utils/cpp_extension.py:782: UserWarning: The detected CUDA version (10.1) has a minor version mismatch with the version that was used to compile PyTorch (10.2). Most likely this shouldn't be a problem.
    warnings.warn(CUDA_MISMATCH_WARN.format(cuda_str_version, torch.version.cuda))
  building 'torch_scatter._scatter_cpu' extension
  creating build/temp.linux-x86_64-3.9
  creating build/temp.linux-x86_64-3.9/csrc
  creating build/temp.linux-x86_64-3.9/csrc/cpu
  gcc -pthread -B /home/abose/anaconda3/envs/pyTorch_Geo/compiler_compat -Wno-unused-result -Wsign-compare -DNDEBUG -O2 -Wall -fPIC -O2 -isystem /home/abose/anaconda3/envs/pyTorch_Geo/include -I/home/abose/anaconda3/envs/pyTorch_Geo/include -fPIC -O2 -isystem /home/abose/anaconda3/envs/pyTorch_Geo/include -fPIC -Icsrc -I/home/abose/anaconda3/envs/pyTorch_Geo/lib/python3.9/site-packages/torch/include -I/home/abose/anaconda3/envs/pyTorch_Geo/lib/python3.9/site-packages/torch/include/torch/csrc/api/include -I/home/abose/anaconda3/envs/pyTorch_Geo/lib/python3.9/site-packages/torch/include/TH -I/home/abose/anaconda3/envs/pyTorch_Geo/lib/python3.9/site-packages/torch/include/THC -I/home/abose/anaconda3/envs/pyTorch_Geo/include/python3.9 -c csrc/cpu/scatter_cpu.cpp -o build/temp.linux-x86_64-3.9/csrc/cpu/scatter_cpu.o -O2 -DAT_PARALLEL_OPENMP -fopenmp -DTORCH_API_INCLUDE_EXTENSION_H -DPYBIND11_COMPILER_TYPE=\"_gcc\" -DPYBIND11_STDLIB=\"_libstdcpp\" -DPYBIND11_BUILD_ABI=\"_cxxabi1011\" -DTORCH_EXTENSION_NAME=_scatter_cpu -D_GLIBCXX_USE_CXX11_ABI=0 -std=c++14
  gcc -pthread -B /home/abose/anaconda3/envs/pyTorch_Geo/compiler_compat -Wno-unused-result -Wsign-compare -DNDEBUG -O2 -Wall -fPIC -O2 -isystem /home/abose/anaconda3/envs/pyTorch_Geo/include -I/home/abose/anaconda3/envs/pyTorch_Geo/include -fPIC -O2 -isystem /home/abose/anaconda3/envs/pyTorch_Geo/include -fPIC -Icsrc -I/home/abose/anaconda3/envs/pyTorch_Geo/lib/python3.9/site-packages/torch/include -I/home/abose/anaconda3/envs/pyTorch_Geo/lib/python3.9/site-packages/torch/include/torch/csrc/api/include -I/home/abose/anaconda3/envs/pyTorch_Geo/lib/python3.9/site-packages/torch/include/TH -I/home/abose/anaconda3/envs/pyTorch_Geo/lib/python3.9/site-packages/torch/include/THC -I/home/abose/anaconda3/envs/pyTorch_Geo/include/python3.9 -c csrc/scatter.cpp -o build/temp.linux-x86_64-3.9/csrc/scatter.o -O2 -DAT_PARALLEL_OPENMP -fopenmp -DTORCH_API_INCLUDE_EXTENSION_H -DPYBIND11_COMPILER_TYPE=\"_gcc\" -DPYBIND11_STDLIB=\"_libstdcpp\" -DPYBIND11_BUILD_ABI=\"_cxxabi1011\" -DTORCH_EXTENSION_NAME=_scatter_cpu -D_GLIBCXX_USE_CXX11_ABI=0 -std=c++14
  g++ -pthread -B /home/abose/anaconda3/envs/pyTorch_Geo/compiler_compat -shared -Wl,-rpath,/home/abose/anaconda3/envs/pyTorch_Geo/lib -Wl,-rpath-link,/home/abose/anaconda3/envs/pyTorch_Geo/lib -L/home/abose/anaconda3/envs/pyTorch_Geo/lib -L/home/abose/anaconda3/envs/pyTorch_Geo/lib -Wl,-rpath,/home/abose/anaconda3/envs/pyTorch_Geo/lib -Wl,-rpath-link,/home/abose/anaconda3/envs/pyTorch_Geo/lib -L/home/abose/anaconda3/envs/pyTorch_Geo/lib build/temp.linux-x86_64-3.9/csrc/cpu/scatter_cpu.o build/temp.linux-x86_64-3.9/csrc/scatter.o -L/home/abose/anaconda3/envs/pyTorch_Geo/lib/python3.9/site-packages/torch/lib -lc10 -ltorch -ltorch_cpu -ltorch_python -o build/lib.linux-x86_64-3.9/torch_scatter/_scatter_cpu.so -s
  building 'torch_scatter._scatter_cuda' extension
  creating build/temp.linux-x86_64-3.9/csrc/cuda
  gcc -pthread -B /home/abose/anaconda3/envs/pyTorch_Geo/compiler_compat -Wno-unused-result -Wsign-compare -DNDEBUG -O2 -Wall -fPIC -O2 -isystem /home/abose/anaconda3/envs/pyTorch_Geo/include -I/home/abose/anaconda3/envs/pyTorch_Geo/include -fPIC -O2 -isystem /home/abose/anaconda3/envs/pyTorch_Geo/include -fPIC -DWITH_CUDA -Icsrc -I/home/abose/anaconda3/envs/pyTorch_Geo/lib/python3.9/site-packages/torch/include -I/home/abose/anaconda3/envs/pyTorch_Geo/lib/python3.9/site-packages/torch/include/torch/csrc/api/include -I/home/abose/anaconda3/envs/pyTorch_Geo/lib/python3.9/site-packages/torch/include/TH -I/home/abose/anaconda3/envs/pyTorch_Geo/lib/python3.9/site-packages/torch/include/THC -I/usr/local/cuda-10.1/include -I/home/abose/anaconda3/envs/pyTorch_Geo/include/python3.9 -c csrc/cpu/scatter_cpu.cpp -o build/temp.linux-x86_64-3.9/csrc/cpu/scatter_cpu.o -O2 -DAT_PARALLEL_OPENMP -fopenmp -DTORCH_API_INCLUDE_EXTENSION_H -DPYBIND11_COMPILER_TYPE=\"_gcc\" -DPYBIND11_STDLIB=\"_libstdcpp\" -DPYBIND11_BUILD_ABI=\"_cxxabi1011\" -DTORCH_EXTENSION_NAME=_scatter_cuda -D_GLIBCXX_USE_CXX11_ABI=0 -std=c++14
  /usr/local/cuda-10.1/bin/nvcc -DWITH_CUDA -Icsrc -I/home/abose/anaconda3/envs/pyTorch_Geo/lib/python3.9/site-packages/torch/include -I/home/abose/anaconda3/envs/pyTorch_Geo/lib/python3.9/site-packages/torch/include/torch/csrc/api/include -I/home/abose/anaconda3/envs/pyTorch_Geo/lib/python3.9/site-packages/torch/include/TH -I/home/abose/anaconda3/envs/pyTorch_Geo/lib/python3.9/site-packages/torch/include/THC -I/usr/local/cuda-10.1/include -I/home/abose/anaconda3/envs/pyTorch_Geo/include/python3.9 -c csrc/cuda/scatter_cuda.cu -o build/temp.linux-x86_64-3.9/csrc/cuda/scatter_cuda.o -D__CUDA_NO_HALF_OPERATORS__ -D__CUDA_NO_HALF_CONVERSIONS__ -D__CUDA_NO_BFLOAT16_CONVERSIONS__ -D__CUDA_NO_HALF2_OPERATORS__ --expt-relaxed-constexpr --compiler-options '-fPIC' --expt-relaxed-constexpr -O2 -DTORCH_API_INCLUDE_EXTENSION_H -DPYBIND11_COMPILER_TYPE=\"_gcc\" -DPYBIND11_STDLIB=\"_libstdcpp\" -DPYBIND11_BUILD_ABI=\"_cxxabi1011\" -DTORCH_EXTENSION_NAME=_scatter_cuda -D_GLIBCXX_USE_CXX11_ABI=0 -gencode=arch=compute_75,code=compute_75 -gencode=arch=compute_75,code=sm_75 -std=c++14
  In file included from csrc/cuda/scatter_cuda.cu:3:0:
  /home/abose/anaconda3/envs/pyTorch_Geo/lib/python3.9/site-packages/torch/include/ATen/cuda/CUDAContext.h:7:10: fatal error: cublas_v2.h: No such file or directory
   #include <cublas_v2.h>
            ^~~~~~~~~~~~~
  compilation terminated.
  error: command '/usr/local/cuda-10.1/bin/nvcc' failed with exit code 1
  ----------------------------------------
  ERROR: Failed building wheel for torch-scatter
  Running setup.py clean for torch-scatter
Failed to build torch-scatter
Installing collected packages: torch-scatter
    Running setup.py install for torch-scatter ... error
    ERROR: Command errored out with exit status 1:
     command: /home/abose/anaconda3/envs/pyTorch_Geo/bin/python -u -c 'import io, os, sys, setuptools, tokenize; sys.argv[0] = '"'"'/tmp/pip-install-k2o__zde/torch-scatter_fd431bdb352b4ae3bb0cc51fb985c3d9/setup.py'"'"'; __file__='"'"'/tmp/pip-install-k2o__zde/torch-scatter_fd431bdb352b4ae3bb0cc51fb985c3d9/setup.py'"'"';f = getattr(tokenize, '"'"'open'"'"', open)(__file__) if os.path.exists(__file__) else io.StringIO('"'"'from setuptools import setup; setup()'"'"');code = f.read().replace('"'"'\r\n'"'"', '"'"'\n'"'"');f.close();exec(compile(code, __file__, '"'"'exec'"'"'))' install --record /tmp/pip-record-kzi86p0e/install-record.txt --single-version-externally-managed --compile --install-headers /home/abose/anaconda3/envs/pyTorch_Geo/include/python3.9/torch-scatter
         cwd: /tmp/pip-install-k2o__zde/torch-scatter_fd431bdb352b4ae3bb0cc51fb985c3d9/
    Complete output (41 lines):
    /home/abose/anaconda3/envs/pyTorch_Geo/lib/python3.9/site-packages/setuptools/dist.py:757: UserWarning: Usage of dash-separated 'description-file' will not be supported in future versions. Please use the underscore name 'description_file' instead
      warnings.warn(
    running install
    /home/abose/anaconda3/envs/pyTorch_Geo/lib/python3.9/site-packages/setuptools/command/install.py:34: SetuptoolsDeprecationWarning: setup.py install is deprecated. Use build and pip and other standards-based tools.
      warnings.warn(
    running build
    running build_py
    creating build
    creating build/lib.linux-x86_64-3.9
    creating build/lib.linux-x86_64-3.9/torch_scatter
    copying torch_scatter/segment_coo.py -> build/lib.linux-x86_64-3.9/torch_scatter
    copying torch_scatter/placeholder.py -> build/lib.linux-x86_64-3.9/torch_scatter
    copying torch_scatter/scatter.py -> build/lib.linux-x86_64-3.9/torch_scatter
    copying torch_scatter/segment_csr.py -> build/lib.linux-x86_64-3.9/torch_scatter
    copying torch_scatter/__init__.py -> build/lib.linux-x86_64-3.9/torch_scatter
    copying torch_scatter/utils.py -> build/lib.linux-x86_64-3.9/torch_scatter
    creating build/lib.linux-x86_64-3.9/torch_scatter/composite
    copying torch_scatter/composite/std.py -> build/lib.linux-x86_64-3.9/torch_scatter/composite
    copying torch_scatter/composite/__init__.py -> build/lib.linux-x86_64-3.9/torch_scatter/composite
    copying torch_scatter/composite/softmax.py -> build/lib.linux-x86_64-3.9/torch_scatter/composite
    copying torch_scatter/composite/logsumexp.py -> build/lib.linux-x86_64-3.9/torch_scatter/composite
    running build_ext
    /home/abose/anaconda3/envs/pyTorch_Geo/lib/python3.9/site-packages/torch/utils/cpp_extension.py:782: UserWarning: The detected CUDA version (10.1) has a minor version mismatch with the version that was used to compile PyTorch (10.2). Most likely this shouldn't be a problem.
      warnings.warn(CUDA_MISMATCH_WARN.format(cuda_str_version, torch.version.cuda))
    building 'torch_scatter._scatter_cpu' extension
    creating build/temp.linux-x86_64-3.9
    creating build/temp.linux-x86_64-3.9/csrc
    creating build/temp.linux-x86_64-3.9/csrc/cpu
    gcc -pthread -B /home/abose/anaconda3/envs/pyTorch_Geo/compiler_compat -Wno-unused-result -Wsign-compare -DNDEBUG -O2 -Wall -fPIC -O2 -isystem /home/abose/anaconda3/envs/pyTorch_Geo/include -I/home/abose/anaconda3/envs/pyTorch_Geo/include -fPIC -O2 -isystem /home/abose/anaconda3/envs/pyTorch_Geo/include -fPIC -Icsrc -I/home/abose/anaconda3/envs/pyTorch_Geo/lib/python3.9/site-packages/torch/include -I/home/abose/anaconda3/envs/pyTorch_Geo/lib/python3.9/site-packages/torch/include/torch/csrc/api/include -I/home/abose/anaconda3/envs/pyTorch_Geo/lib/python3.9/site-packages/torch/include/TH -I/home/abose/anaconda3/envs/pyTorch_Geo/lib/python3.9/site-packages/torch/include/THC -I/home/abose/anaconda3/envs/pyTorch_Geo/include/python3.9 -c csrc/cpu/scatter_cpu.cpp -o build/temp.linux-x86_64-3.9/csrc/cpu/scatter_cpu.o -O2 -DAT_PARALLEL_OPENMP -fopenmp -DTORCH_API_INCLUDE_EXTENSION_H -DPYBIND11_COMPILER_TYPE=\"_gcc\" -DPYBIND11_STDLIB=\"_libstdcpp\" -DPYBIND11_BUILD_ABI=\"_cxxabi1011\" -DTORCH_EXTENSION_NAME=_scatter_cpu -D_GLIBCXX_USE_CXX11_ABI=0 -std=c++14
    gcc -pthread -B /home/abose/anaconda3/envs/pyTorch_Geo/compiler_compat -Wno-unused-result -Wsign-compare -DNDEBUG -O2 -Wall -fPIC -O2 -isystem /home/abose/anaconda3/envs/pyTorch_Geo/include -I/home/abose/anaconda3/envs/pyTorch_Geo/include -fPIC -O2 -isystem /home/abose/anaconda3/envs/pyTorch_Geo/include -fPIC -Icsrc -I/home/abose/anaconda3/envs/pyTorch_Geo/lib/python3.9/site-packages/torch/include -I/home/abose/anaconda3/envs/pyTorch_Geo/lib/python3.9/site-packages/torch/include/torch/csrc/api/include -I/home/abose/anaconda3/envs/pyTorch_Geo/lib/python3.9/site-packages/torch/include/TH -I/home/abose/anaconda3/envs/pyTorch_Geo/lib/python3.9/site-packages/torch/include/THC -I/home/abose/anaconda3/envs/pyTorch_Geo/include/python3.9 -c csrc/scatter.cpp -o build/temp.linux-x86_64-3.9/csrc/scatter.o -O2 -DAT_PARALLEL_OPENMP -fopenmp -DTORCH_API_INCLUDE_EXTENSION_H -DPYBIND11_COMPILER_TYPE=\"_gcc\" -DPYBIND11_STDLIB=\"_libstdcpp\" -DPYBIND11_BUILD_ABI=\"_cxxabi1011\" -DTORCH_EXTENSION_NAME=_scatter_cpu -D_GLIBCXX_USE_CXX11_ABI=0 -std=c++14
    g++ -pthread -B /home/abose/anaconda3/envs/pyTorch_Geo/compiler_compat -shared -Wl,-rpath,/home/abose/anaconda3/envs/pyTorch_Geo/lib -Wl,-rpath-link,/home/abose/anaconda3/envs/pyTorch_Geo/lib -L/home/abose/anaconda3/envs/pyTorch_Geo/lib -L/home/abose/anaconda3/envs/pyTorch_Geo/lib -Wl,-rpath,/home/abose/anaconda3/envs/pyTorch_Geo/lib -Wl,-rpath-link,/home/abose/anaconda3/envs/pyTorch_Geo/lib -L/home/abose/anaconda3/envs/pyTorch_Geo/lib build/temp.linux-x86_64-3.9/csrc/cpu/scatter_cpu.o build/temp.linux-x86_64-3.9/csrc/scatter.o -L/home/abose/anaconda3/envs/pyTorch_Geo/lib/python3.9/site-packages/torch/lib -lc10 -ltorch -ltorch_cpu -ltorch_python -o build/lib.linux-x86_64-3.9/torch_scatter/_scatter_cpu.so -s
    building 'torch_scatter._scatter_cuda' extension
    creating build/temp.linux-x86_64-3.9/csrc/cuda
    gcc -pthread -B /home/abose/anaconda3/envs/pyTorch_Geo/compiler_compat -Wno-unused-result -Wsign-compare -DNDEBUG -O2 -Wall -fPIC -O2 -isystem /home/abose/anaconda3/envs/pyTorch_Geo/include -I/home/abose/anaconda3/envs/pyTorch_Geo/include -fPIC -O2 -isystem /home/abose/anaconda3/envs/pyTorch_Geo/include -fPIC -DWITH_CUDA -Icsrc -I/home/abose/anaconda3/envs/pyTorch_Geo/lib/python3.9/site-packages/torch/include -I/home/abose/anaconda3/envs/pyTorch_Geo/lib/python3.9/site-packages/torch/include/torch/csrc/api/include -I/home/abose/anaconda3/envs/pyTorch_Geo/lib/python3.9/site-packages/torch/include/TH -I/home/abose/anaconda3/envs/pyTorch_Geo/lib/python3.9/site-packages/torch/include/THC -I/usr/local/cuda-10.1/include -I/home/abose/anaconda3/envs/pyTorch_Geo/include/python3.9 -c csrc/cpu/scatter_cpu.cpp -o build/temp.linux-x86_64-3.9/csrc/cpu/scatter_cpu.o -O2 -DAT_PARALLEL_OPENMP -fopenmp -DTORCH_API_INCLUDE_EXTENSION_H -DPYBIND11_COMPILER_TYPE=\"_gcc\" -DPYBIND11_STDLIB=\"_libstdcpp\" -DPYBIND11_BUILD_ABI=\"_cxxabi1011\" -DTORCH_EXTENSION_NAME=_scatter_cuda -D_GLIBCXX_USE_CXX11_ABI=0 -std=c++14
    /usr/local/cuda-10.1/bin/nvcc -DWITH_CUDA -Icsrc -I/home/abose/anaconda3/envs/pyTorch_Geo/lib/python3.9/site-packages/torch/include -I/home/abose/anaconda3/envs/pyTorch_Geo/lib/python3.9/site-packages/torch/include/torch/csrc/api/include -I/home/abose/anaconda3/envs/pyTorch_Geo/lib/python3.9/site-packages/torch/include/TH -I/home/abose/anaconda3/envs/pyTorch_Geo/lib/python3.9/site-packages/torch/include/THC -I/usr/local/cuda-10.1/include -I/home/abose/anaconda3/envs/pyTorch_Geo/include/python3.9 -c csrc/cuda/scatter_cuda.cu -o build/temp.linux-x86_64-3.9/csrc/cuda/scatter_cuda.o -D__CUDA_NO_HALF_OPERATORS__ -D__CUDA_NO_HALF_CONVERSIONS__ -D__CUDA_NO_BFLOAT16_CONVERSIONS__ -D__CUDA_NO_HALF2_OPERATORS__ --expt-relaxed-constexpr --compiler-options '-fPIC' --expt-relaxed-constexpr -O2 -DTORCH_API_INCLUDE_EXTENSION_H -DPYBIND11_COMPILER_TYPE=\"_gcc\" -DPYBIND11_STDLIB=\"_libstdcpp\" -DPYBIND11_BUILD_ABI=\"_cxxabi1011\" -DTORCH_EXTENSION_NAME=_scatter_cuda -D_GLIBCXX_USE_CXX11_ABI=0 -gencode=arch=compute_75,code=compute_75 -gencode=arch=compute_75,code=sm_75 -std=c++14
    In file included from csrc/cuda/scatter_cuda.cu:3:0:
    /home/abose/anaconda3/envs/pyTorch_Geo/lib/python3.9/site-packages/torch/include/ATen/cuda/CUDAContext.h:7:10: fatal error: cublas_v2.h: No such file or directory
     #include <cublas_v2.h>
              ^~~~~~~~~~~~~
    compilation terminated.
    error: command '/usr/local/cuda-10.1/bin/nvcc' failed with exit code 1
    ----------------------------------------
ERROR: Command errored out with exit status 1: /home/abose/anaconda3/envs/pyTorch_Geo/bin/python -u -c 'import io, os, sys, setuptools, tokenize; sys.argv[0] = '"'"'/tmp/pip-install-k2o__zde/torch-scatter_fd431bdb352b4ae3bb0cc51fb985c3d9/setup.py'"'"'; __file__='"'"'/tmp/pip-install-k2o__zde/torch-scatter_fd431bdb352b4ae3bb0cc51fb985c3d9/setup.py'"'"';f = getattr(tokenize, '"'"'open'"'"', open)(__file__) if os.path.exists(__file__) else io.StringIO('"'"'from setuptools import setup; setup()'"'"');code = f.read().replace('"'"'\r\n'"'"', '"'"'\n'"'"');f.close();exec(compile(code, __file__, '"'"'exec'"'"'))' install --record /tmp/pip-record-kzi86p0e/install-record.txt --single-version-externally-managed --compile --install-headers /home/abose/anaconda3/envs/pyTorch_Geo/include/python3.9/torch-scatter Check the logs for full command output.

[rusty1s]

You need to set ${CUDA} to the value of torch.version.cuda, e.g.:

pip install torch-scatter -f https://data.pyg.org/whl/torch-1.10.0+cu102.html

[abose17]

Thanks a lot, Matthias. It worked after a new installation.