Installation on aarch64 ubuntu 20.04 failed due to `unknown architectural extension flagm`

[richgrove]

🐛 Describe the bug

Run the installation:

$ pip3 install pyg-lib -f https://data.pyg.org/whl/torch-2.0.0+cu118.html Looking in links: https://data.pyg.org/whl/torch-2.0.0+cu118.html ERROR: Could not find a version that satisfies the requirement pyg-lib (from versions: none) ERROR: No matching distribution found for pyg-lib

This is on Arm64. The PyTorch 2.0.0 installed with Cuda 11.8 works fine. If it's not supported, please add it as a feature request. Thanks.

Environment

• pyg-lib version:
• PyTorch version:
• OS: ubuntu 20.04 arm64
• Python version: 3.8
• CUDA/cuDNN version: 11.8
• How you installed PyTorch and pyg-lib (conda, pip, source): pip3 install torch-2.0.0+nv23.05-cp38-cp38-linux_aarch64.whl
• Any other relevant information:

[richgrove]

Here are the error messages when ran:

$ pip3 install git+https://github.com/pyg-team/pyg-lib.git
Collecting git+https://github.com/pyg-team/pyg-lib.git
  Cloning https://github.com/pyg-team/pyg-lib.git to /tmp/pip-req-build-7trih_7i
  Running command git clone --filter=blob:none --quiet https://github.com/pyg-team/pyg-lib.git /tmp/pip-req-build-7trih_7i
  Resolved https://github.com/pyg-team/pyg-lib.git to commit f90a23169779fc372a94f0b8bcd424890a9bdd35
  Running command git submodule update --init --recursive -q
  Preparing metadata (setup.py) ... done
Building wheels for collected packages: pyg-lib
  Building wheel for pyg-lib (setup.py) ... error
  error: subprocess-exited-with-error

  × python setup.py bdist_wheel did not run successfully.
  │ exit code: 1
  ╰─> [244 lines of output]
      running bdist_wheel
      running build
      running build_py
      creating build
      creating build/lib.linux-aarch64-cpython-38
      creating build/lib.linux-aarch64-cpython-38/pyg_lib
      copying pyg_lib/_triton.py -> build/lib.linux-aarch64-cpython-38/pyg_lib
      copying pyg_lib/__init__.py -> build/lib.linux-aarch64-cpython-38/pyg_lib
      copying pyg_lib/testing.py -> build/lib.linux-aarch64-cpython-38/pyg_lib
      copying pyg_lib/home.py -> build/lib.linux-aarch64-cpython-38/pyg_lib
      creating build/lib.linux-aarch64-cpython-38/pyg_lib/ops
      copying pyg_lib/ops/__init__.py -> build/lib.linux-aarch64-cpython-38/pyg_lib/ops
      copying pyg_lib/ops/scatter_reduce.py -> build/lib.linux-aarch64-cpython-38/pyg_lib/ops
      creating build/lib.linux-aarch64-cpython-38/pyg_lib/partition
      copying pyg_lib/partition/__init__.py -> build/lib.linux-aarch64-cpython-38/pyg_lib/partition
      creating build/lib.linux-aarch64-cpython-38/pyg_lib/sampler
      copying pyg_lib/sampler/__init__.py -> build/lib.linux-aarch64-cpython-38/pyg_lib/sampler
      running build_ext

      -- The C compiler identification is GNU 11.4.0
      -- The CXX compiler identification is GNU 11.4.0
      -- Detecting C compiler ABI info
      -- Detecting C compiler ABI info - done
      -- Check for working C compiler: /usr/bin/cc - skipped
      -- Detecting C compile features
      -- Detecting C compile features - done
      -- Detecting CXX compiler ABI info
      -- Detecting CXX compiler ABI info - done
      -- Check for working CXX compiler: /usr/bin/c++ - skipped
      -- Detecting CXX compile features
      -- Detecting CXX compile features - done
      -- Found Python3: /xxxx/anaconda3/envs/dev/include/python3.8 (found version "3.8.18") found components: Development Development.Module Development.Embed
      -- The CUDA compiler identification is NVIDIA 11.8.89
      -- Detecting CUDA compiler ABI info
      -- Detecting CUDA compiler ABI info - done
      -- Check for working CUDA compiler: /usr/local/cuda/bin/nvcc - skipped
      -- Detecting CUDA compile features
      -- Detecting CUDA compile features - done
      -- Looking for execinfo.h
      -- Looking for execinfo.h - found
      -- Looking for getline
      -- Looking for getline - found
      -- Found CUDA: /usr/local/cuda (found version "11.8")
      -- Caffe2: CUDA detected: 11.8
      -- Caffe2: CUDA nvcc is: /usr/local/cuda/bin/nvcc
      -- Caffe2: CUDA toolkit directory: /usr/local/cuda
      -- Caffe2: Header version is: 11.8
      -- /usr/local/cuda/lib64/libnvrtc.so shorthash is cb86903b
      -- USE_CUDNN is set to 0. Compiling without cuDNN support
      -- Autodetected CUDA architecture(s):  8.7
      -- Added CUDA NVCC flags for: -gencode;arch=compute_87,code=sm_87
      -- Found Torch: /xxxx/anaconda3/envs/dev/lib/python3.8/site-packages/torch/lib/libtorch.so
      -- Could NOT find OpenMP_C (missing: OpenMP_C_FLAGS OpenMP_C_LIB_NAMES)
      -- Found OpenMP_CXX: -fopenmp (found version "4.5")
      -- Could NOT find OpenMP (missing: OpenMP_C_FOUND) (found version "4.5")
      -- CMAKE_BUILD_TYPE: RELEASE
      -- CMAKE_CXX_FLAGS_DEBUG: -g -O0
      -- CMAKE_CXX_FLAGS_RELEASE: -O3 -DNDEBUG
      -- Configuring done (9.6s)
      CMake Warning (dev) in CMakeLists.txt:
        Policy CMP0104 is not set: CMAKE_CUDA_ARCHITECTURES now detected for NVCC,
        empty CUDA_ARCHITECTURES not allowed.  Run "cmake --help-policy CMP0104"
        for policy details.  Use the cmake_policy command to set the policy and
        suppress this warning.
     
        CUDA_ARCHITECTURES is empty for target pyg
      This warning is for project developers.  Use -Wno-dev to suppress it.
      
      CMake Warning at CMakeLists.txt:55 (add_library):
        Cannot generate a safe runtime search path for target pyg because files in
        some directories may conflict with libraries in implicit directories:
      
          runtime library [libgomp.so.1] in /usr/lib/gcc/aarch64-linux-gnu/11 may be hidden by files in:
            /xxxx/anaconda3/envs/dev/lib
      
        Some of these libraries may not be found correctly.
      
      
      -- Generating done (0.0s)
      -- Build files have been written to: /tmp/pip-req-build-7trih_7i/build/temp.linux-aarch64-cpython-38
      [1/92] Building CXX object CMakeFiles/pyg.dir/pyg_lib/csrc/library.cpp.o
      <command-line>: warning: "_GLIBCXX_USE_CXX11_ABI" redefined
      <command-line>: note: this is the location of the previous definition
      In file included from /tmp/pip-req-build-7trih_7i/pyg_lib/csrc/library.cpp:1:
      /tmp/pip-req-build-7trih_7i/pyg_lib/csrc/library.h:12:37: warning: ‘_register_ops’ initialized and declared ‘extern’
         12 | extern "C" PYG_INLINE_VARIABLE auto _register_ops = &cuda_version;
            |                                     ^~~~~~~~~~~~~
      [2/92] Building CXX object CMakeFiles/pyg.dir/pyg_lib/csrc/ops/cpu/sampled_kernel.cpp.o
      <command-line>: warning: "_GLIBCXX_USE_CXX11_ABI" redefined
      <command-line>: note: this is the location of the previous definition
      [3/92] Building CXX object CMakeFiles/pyg.dir/pyg_lib/csrc/ops/cpu/index_sort_kernel.cpp.o
      <command-line>: warning: "_GLIBCXX_USE_CXX11_ABI" redefined
      <command-line>: note: this is the location of the previous definition
      [4/92] Building CXX object CMakeFiles/pyg.dir/pyg_lib/csrc/ops/cpu/softmax_kernel.cpp.o
      <command-line>: warning: "_GLIBCXX_USE_CXX11_ABI" redefined
      <command-line>: note: this is the location of the previous definition
      [5/92] Building CXX object CMakeFiles/pyg.dir/pyg_lib/csrc/ops/cpu/matmul_kernel.cpp.o
      <command-line>: warning: "_GLIBCXX_USE_CXX11_ABI" redefined
      <command-line>: note: this is the location of the previous definition
      [6/92] Building CXX object CMakeFiles/pyg.dir/pyg_lib/csrc/ops/index_sort.cpp.o
      <command-line>: warning: "_GLIBCXX_USE_CXX11_ABI" redefined
      <command-line>: note: this is the location of the previous definition
      [7/92] Building CXX object CMakeFiles/pyg.dir/pyg_lib/csrc/ops/autograd/softmax_kernel.cpp.o
      <command-line>: warning: "_GLIBCXX_USE_CXX11_ABI" redefined
      <command-line>: note: this is the location of the previous definition
      [8/92] Building CXX object CMakeFiles/pyg.dir/pyg_lib/csrc/ops/autograd/sampled_kernel.cpp.o
      <command-line>: warning: "_GLIBCXX_USE_CXX11_ABI" redefined
      <command-line>: note: this is the location of the previous definition
      [9/92] Building CXX object CMakeFiles/pyg.dir/pyg_lib/csrc/ops/autograd/matmul_kernel.cpp.o
      <command-line>: warning: "_GLIBCXX_USE_CXX11_ABI" redefined
      <command-line>: note: this is the location of the previous definition
      [10/92] Building CXX object CMakeFiles/pyg.dir/pyg_lib/csrc/random/cpu/biased_sampling.cpp.o
      <command-line>: warning: "_GLIBCXX_USE_CXX11_ABI" redefined
      <command-line>: note: this is the location of the previous definition
      [11/92] Building CXX object CMakeFiles/pyg.dir/pyg_lib/csrc/ops/sampled.cpp.o
      <command-line>: warning: "_GLIBCXX_USE_CXX11_ABI" redefined
      <command-line>: note: this is the location of the previous definition
      [12/92] Building CXX object CMakeFiles/pyg.dir/pyg_lib/csrc/partition/cpu/metis_kernel.cpp.o
      <command-line>: warning: "_GLIBCXX_USE_CXX11_ABI" redefined
      <command-line>: note: this is the location of the previous definition
      [13/92] Building CXX object CMakeFiles/pyg.dir/pyg_lib/csrc/partition/metis.cpp.o
      <command-line>: warning: "_GLIBCXX_USE_CXX11_ABI" redefined
      <command-line>: note: this is the location of the previous definition
      [14/92] Building CXX object CMakeFiles/pyg.dir/pyg_lib/csrc/ops/softmax.cpp.o
      <command-line>: warning: "_GLIBCXX_USE_CXX11_ABI" redefined
      <command-line>: note: this is the location of the previous definition
      [15/92] Building CXX object CMakeFiles/pyg.dir/pyg_lib/csrc/ops/matmul.cpp.o
      <command-line>: warning: "_GLIBCXX_USE_CXX11_ABI" redefined
      <command-line>: note: this is the location of the previous definition
      [16/92] Building CXX object CMakeFiles/pyg.dir/pyg_lib/csrc/sampler/cpu/dist_merge_outputs_kernel.cpp.o
      <command-line>: warning: "_GLIBCXX_USE_CXX11_ABI" redefined
      <command-line>: note: this is the location of the previous definition
      /tmp/pip-req-build-7trih_7i/pyg_lib/csrc/sampler/cpu/dist_merge_outputs_kernel.cpp: In function std::tuple<at::Tensor, at::Tensor, c10::optional<at::Tensor>, std::vector<long int, std::allocator<long int> > > pyg::sampler::merge_sampler_outputs_kernel(const std::vector<at::Tensor>&, const std::vector<at::Tensor>&, const std::vector<std::vector<long int> >&, const std::vector<long int>&, const std::vector<long int>&, int64_t, int64_t, const c10::optional<at::Tensor>&, bool):
      /tmp/pip-req-build-7trih_7i/pyg_lib/csrc/sampler/cpu/dist_merge_outputs_kernel.cpp:159:1: warning: control reaches end of non-void function [-Wreturn-type]
        159 | }
            | ^
      [17/92] Building CXX object CMakeFiles/pyg.dir/pyg_lib/csrc/sampler/cpu/random_walk_kernel.cpp.o
      <command-line>: warning: "_GLIBCXX_USE_CXX11_ABI" redefined
      <command-line>: note: this is the location of the previous definition
      [18/92] Building CXX object CMakeFiles/pyg.dir/pyg_lib/csrc/sampler/dist_merge_outputs.cpp.o
      <command-line>: warning: "_GLIBCXX_USE_CXX11_ABI" redefined
      <command-line>: note: this is the location of the previous definition
      [19/92] Building CUDA object CMakeFiles/pyg.dir/pyg_lib/csrc/ops/cuda/sampled_kernel.cu.o
      [20/92] Building CXX object CMakeFiles/pyg.dir/pyg_lib/csrc/sampler/cpu/subgraph_kernel.cpp.o
      <command-line>: warning: "_GLIBCXX_USE_CXX11_ABI" redefined
      <command-line>: note: this is the location of the previous definition
      [21/92] Building CXX object CMakeFiles/pyg.dir/pyg_lib/csrc/sampler/dist_relabel.cpp.o
      <command-line>: warning: "_GLIBCXX_USE_CXX11_ABI" redefined
      <command-line>: note: this is the location of the previous definition
      [22/92] Building C object third_party/METIS/libmetis/CMakeFiles/metis.dir/__/GKlib/b64.c.o
      FAILED: third_party/METIS/libmetis/CMakeFiles/metis.dir/__/GKlib/b64.c.o
      /usr/bin/cc -DUSE_PYTHON -DWITH_CUDA -I/tmp/pip-req-build-7trih_7i/third_party/cutlass/include -I/tmp/pip-req-build-7trih_7i/third_party/cutlass/tools/util/include -I/tmp/pip-req-build-7trih_7i/third_party/METIS/GKlib -I/tmp/pip-req-build-7trih_7i/third_party/METIS/include -I/tmp/pip-req-build-7trih_7i/third_party/METIS/libmetis/. -DIDXTYPEWIDTH=64 -DREALTYPEWIDTH=32 -DLINUX -D_FILE_OFFSET_BITS=64 -std=c99 -fno-strict-aliasing -march=native -fPIC -Werror -Wall -pedantic -Wno-unused-function -Wno-unused-but-set-variable -Wno-unused-variable -Wno-unknown-pragmas -Wno-unused-label -DNDEBUG -DNDEBUG2 -DHAVE_EXECINFO_H -DHAVE_GETLINE -O3 -O3 -DNDEBUG -MD -MT third_party/METIS/libmetis/CMakeFiles/metis.dir/__/GKlib/b64.c.o -MF third_party/METIS/libmetis/CMakeFiles/metis.dir/__/GKlib/b64.c.o.d -o third_party/METIS/libmetis/CMakeFiles/metis.dir/__/GKlib/b64.c.o -c /tmp/pip-req-build-7trih_7i/third_party/METIS/GKlib/b64.c
      Assembler messages:
      Error: unknown architectural extension flagm
      Error: unrecognized option -march=armv8.2-a+crypto+fp16+rcpc+dotprod+flagm
      [23/92] Building CXX object CMakeFiles/pyg.dir/pyg_lib/csrc/sampler/neighbor.cpp.o
      <command-line>: warning: "_GLIBCXX_USE_CXX11_ABI" redefined
      <command-line>: note: this is the location of the previous definition
      [24/92] Building CXX object CMakeFiles/pyg.dir/pyg_lib/csrc/sampler/random_walk.cpp.o
      <command-line>: warning: "_GLIBCXX_USE_CXX11_ABI" redefined
      <command-line>: note: this is the location of the previous definition
      [25/92] Building CXX object CMakeFiles/pyg.dir/pyg_lib/csrc/utils/check.cpp.o
      <command-line>: warning: "_GLIBCXX_USE_CXX11_ABI" redefined
      <command-line>: note: this is the location of the previous definition
      [26/92] Building CXX object CMakeFiles/pyg.dir/pyg_lib/csrc/utils/convert.cpp.o
      <command-line>: warning: "_GLIBCXX_USE_CXX11_ABI" redefined
      <command-line>: note: this is the location of the previous definition
      [27/92] Building CXX object CMakeFiles/pyg.dir/pyg_lib/csrc/sampler/subgraph.cpp.o
      <command-line>: warning: "_GLIBCXX_USE_CXX11_ABI" redefined
      <command-line>: note: this is the location of the previous definition
      [28/92] Building CXX object CMakeFiles/pyg.dir/pyg_lib/csrc/sampler/cpu/dist_relabel_kernel.cpp.o
      <command-line>: warning: "_GLIBCXX_USE_CXX11_ABI" redefined
      <command-line>: note: this is the location of the previous definition
      /tmp/pip-req-build-7trih_7i/pyg_lib/csrc/sampler/cpu/dist_relabel_kernel.cpp: In function ‘std::tuple<at::Tensor, at::Tensor> pyg::sampler::relabel_neighborhood_kernel(const at::Tensor&, const at::Tensor&, const std::vector<long int>&, int64_t, const c10::optional<at::Tensor>&, bool, bool)’:
      /tmp/pip-req-build-7trih_7i/pyg_lib/csrc/sampler/cpu/dist_relabel_kernel.cpp:290:1: warning: control reaches end of non-void function [-Wreturn-type]
        290 | }
            | ^
      /tmp/pip-req-build-7trih_7i/pyg_lib/csrc/sampler/cpu/dist_relabel_kernel.cpp: In function ‘std::tuple<c10::Dict<std::__cxx11::basic_string<char, std::char_traits<char>, std::allocator<char> >, at::Tensor>, c10::Dict<std::__cxx11::basic_string<char, std::char_traits<char>, std::allocator<char> >, at::Tensor> > pyg::sampler::hetero_relabel_neighborhood_kernel(const std::vector<std::__cxx11::basic_string<char> >&, const std::vector<std::tuple<std::__cxx11::basic_string<char, std::char_traits<char>, std::allocator<char> >, std::__cxx11::basic_string<char, std::char_traits<char>, std::allocator<char> >, std::__cxx11::basic_string<char, std::char_traits<char>, std::allocator<char> > > >&, const c10::Dict<std::__cxx11::basic_string<char>, at::Tensor>&, const c10::Dict<std::__cxx11::basic_string<char>, at::Tensor>&, const c10::Dict<std::__cxx11::basic_string<char>, std::vector<std::vector<long int> > >&, const c10::Dict<std::__cxx11::basic_string<char>, long int>&, const c10::optional<c10::Dict<std::__cxx11::basic_string<char>, at::Tensor> >&, bool, bool)’:
      /tmp/pip-req-build-7trih_7i/pyg_lib/csrc/sampler/cpu/dist_relabel_kernel.cpp:309:1: warning: control reaches end of non-void function [-Wreturn-type]
        309 | }
            | ^
      [29/92] Building CUDA object CMakeFiles/pyg.dir/pyg_lib/csrc/sampler/cuda/random_walk_kernel.cu.o
      [30/92] Building CUDA object CMakeFiles/pyg.dir/pyg_lib/csrc/ops/cuda/matmul_kernel.cu.o
      [31/92] Building CXX object CMakeFiles/pyg.dir/pyg_lib/csrc/sampler/cpu/neighbor_kernel.cpp.o
      <command-line>: warning: "_GLIBCXX_USE_CXX11_ABI" redefined
      <command-line>: note: this is the location of the previous definition
      /tmp/pip-req-build-7trih_7i/pyg_lib/csrc/sampler/cpu/neighbor_kernel.cpp: In lambda function:
      /tmp/pip-req-build-7trih_7i/pyg_lib/csrc/sampler/cpu/neighbor_kernel.cpp:917:3: warning: control reaches end of non-void function [-Wreturn-type]
        917 |   }();
            |   ^
      /tmp/pip-req-build-7trih_7i/pyg_lib/csrc/sampler/cpu/neighbor_kernel.cpp: In function ‘std::tuple<c10::Dict<std::__cxx11::basic_string<char, std::char_traits<char>, std::allocator<char> >, at::Tensor>, c10::Dict<std::__cxx11::basic_string<char, std::char_traits<char>, std::allocator<char> >, at::Tensor>, c10::Dict<std::__cxx11::basic_string<char, std::char_traits<char>, std::allocator<char> >, at::Tensor>, c10::optional<c10::Dict<std::__cxx11::basic_string<char, std::char_traits<char>, std::allocator<char> >, at::Tensor> >, c10::Dict<std::__cxx11::basic_string<char, std::char_traits<char>, std::allocator<char> >, std::vector<long int, std::allocator<long int> > >, c10::Dict<std::__cxx11::basic_string<char, std::char_traits<char>, std::allocator<char> >, std::vector<long int, std::allocator<long int> > > > pyg::sampler::hetero_neighbor_sample_kernel(const std::vector<std::__cxx11::basic_string<char> >&, const std::vector<std::tuple<std::__cxx11::basic_string<char, std::char_traits<char>, std::allocator<char> >, std::__cxx11::basic_string<char, std::char_traits<char>, std::allocator<char> >, std::__cxx11::basic_string<char, std::char_traits<char>, std::allocator<char> > > >&, const c10::Dict<std::__cxx11::basic_string<char>, at::Tensor>&, const c10::Dict<std::__cxx11::basic_string<char>, at::Tensor>&, const c10::Dict<std::__cxx11::basic_string<char>, at::Tensor>&, const c10::Dict<std::__cxx11::basic_string<char>, std::vector<long int> >&, const c10::optional<c10::Dict<std::__cxx11::basic_string<char>, at::Tensor> >&, const c10::optional<c10::Dict<std::__cxx11::basic_string<char>, at::Tensor> >&, const c10::optional<c10::Dict<std::__cxx11::basic_string<char>, at::Tensor> >&, const c10::optional<c10::Dict<std::__cxx11::basic_string<char>, at::Tensor> >&, bool, bool, bool, bool, std::string, bool)’:
      /tmp/pip-req-build-7trih_7i/pyg_lib/csrc/sampler/cpu/neighbor_kernel.cpp:949:1: warning: control reaches end of non-void function [-Wreturn-type]
        949 | }
            | ^
      /tmp/pip-req-build-7trih_7i/pyg_lib/csrc/sampler/cpu/neighbor_kernel.cpp: In lambda function:
      /tmp/pip-req-build-7trih_7i/pyg_lib/csrc/sampler/cpu/neighbor_kernel.cpp:969:3: warning: control reaches end of non-void function [-Wreturn-type]
        969 |   }();
            |   ^
      ninja: build stopped: subcommand failed.
      Traceback (most recent call last):
        File "<string>", line 2, in <module>
        File "<pip-setuptools-caller>", line 34, in <module>
        File "/tmp/pip-req-build-7trih_7i/setup.py", line 129, in <module>
          setup(
        File "/xxxx/anaconda3/envs/dev/lib/python3.8/site-packages/setuptools/__init__.py", line 103, in setup
          return distutils.core.setup(**attrs)
        File "/xxxx/anaconda3/envs/dev/lib/python3.8/site-packages/setuptools/_distutils/core.py", line 185, in setup
          return run_commands(dist)
        File "/xxxx/anaconda3/envs/dev/lib/python3.8/site-packages/setuptools/_distutils/core.py", line 201, in run_commands
          dist.run_commands()
        File "/xxxx/anaconda3/envs/dev/lib/python3.8/site-packages/setuptools/_distutils/dist.py", line 969, in run_commands
          self.run_command(cmd)
        File "/xxxx/anaconda3/envs/dev/lib/python3.8/site-packages/setuptools/dist.py", line 989, in run_command
          super().run_command(command)
        File "/xxxx/anaconda3/envs/dev/lib/python3.8/site-packages/setuptools/_distutils/dist.py", line 988, in run_command
          cmd_obj.run()
        File "/xxxx/anaconda3/envs/dev/lib/python3.8/site-packages/wheel/bdist_wheel.py", line 364, in run
          self.run_command("build")
        File "/xxxx/anaconda3/envs/dev/lib/python3.8/site-packages/setuptools/_distutils/cmd.py", line 318, in run_command
          self.distribution.run_command(command)
        File "/xxxx/anaconda3/envs/dev/lib/python3.8/site-packages/setuptools/dist.py", line 989, in run_command
          super().run_command(command)
        File "/xxxx/anaconda3/envs/dev/lib/python3.8/site-packages/setuptools/_distutils/dist.py", line 988, in run_command
          cmd_obj.run()
        File "/xxxx/anaconda3/envs/dev/lib/python3.8/site-packages/setuptools/_distutils/command/build.py", line 131, in run
          self.run_command(cmd_name)
        File "/xxxx/anaconda3/envs/dev/lib/python3.8/site-packages/setuptools/_distutils/cmd.py", line 318, in run_command
          self.distribution.run_command(command)
        File "/xxxx/anaconda3/envs/dev/lib/python3.8/site-packages/setuptools/dist.py", line 989, in run_command
          super().run_command(command)
        File "/xxxx/anaconda3/envs/dev/lib/python3.8/site-packages/setuptools/_distutils/dist.py", line 988, in run_command
          cmd_obj.run()
        File "/xxxx/anaconda3/envs/dev/lib/python3.8/site-packages/setuptools/command/build_ext.py", line 88, in run
          _build_ext.run(self)
        File "/xxxx/anaconda3/envs/dev/lib/python3.8/site-packages/setuptools/_distutils/command/build_ext.py", line 345, in run
          self.build_extensions()
        File "/xxxx/anaconda3/envs/dev/lib/python3.8/site-packages/setuptools/_distutils/command/build_ext.py", line 467, in build_extensions
          self._build_extensions_serial()
        File "/xxxx/anaconda3/envs/dev/lib/python3.8/site-packages/setuptools/_distutils/command/build_ext.py", line 493, in _build_extensions_serial
          self.build_extension(ext)
        File "/tmp/pip-req-build-7trih_7i/setup.py", line 84, in build_extension
          subprocess.check_call(['cmake', '--build', '.'] + build_args,
        File "/xxxx/anaconda3/envs/dev/lib/python3.8/subprocess.py", line 364, in check_call
          raise CalledProcessError(retcode, cmd)
      subprocess.CalledProcessError: Command '['cmake', '--build', '.']' returned non-zero exit status 1.
      [end of output]
  
  note: This error originates from a subprocess, and is likely not a problem with pip.
  ERROR: Failed building wheel for pyg-lib
  Running setup.py clean for pyg-lib
Failed to build pyg-lib
ERROR: Could not build wheels for pyg-lib, which is required to install pyproject.toml-based projects

[rusty1s]

This looks to be an issue related to CUTLASS, not yet sure how to fix this on our end (cc @puririshi98). In the worst case, try to install via FORCE_CUDA=0 pip install ...