ERROR: Could not build wheels for patrickstar, which is required to install pyproject.toml-based projects

[arminzhu]

I want to reproduce PatrickStar, but I got this. @reyoung

      /usr/local/lib/python3.12/dist-packages/torch/include/torch/csrc/api/include/torch/nn/modules/conv.h: In instantiation of ‘void torch::nn::ConvNdImpl<D, Derived>::reset_parameters() [with long unsigned int D = 3; Derived = torch::nn::Conv3dImpl]’:
      /usr/local/lib/python3.12/dist-packages/torch/include/torch/csrc/api/include/torch/nn/modules/conv.h:100:5:   required from ‘void torch::nn::ConvNdImpl<D, Derived>::reset() [with long unsigned int D = 3; Derived = torch::nn::Conv3dImpl]’
      /usr/local/lib/python3.12/dist-packages/torch/include/torch/csrc/api/include/torch/nn/modules/conv.h:32:8:   required from here
      /usr/local/lib/python3.12/dist-packages/torch/include/torch/csrc/api/include/torch/nn/modules/conv.h:104:27: error: cannot convert ‘const torch::enumtype::kFanIn’ to ‘int’
        104 |     init::kaiming_uniform_(
            |     ~~~~~~~~~~~~~~~~~~~~~~^
            |                           |
            |                           const torch::enumtype::kFanIn
        105 |         weight,
            |         ~~~~~~~
        106 |         /*a=*/std::sqrt(5)); // NOLINT(cppcoreguidelines-avoid-magic-numbers)
            |         ~~~~~~~~~~~~~~~~~~~
      /usr/local/lib/python3.12/dist-packages/torch/include/torch/csrc/api/include/torch/nn/init.h:98:17: note:   initializing argument 3 of ‘at::Tensor torch::nn::init::kaiming_uniform_(at::Tensor, double, int, int)’
         98 |     FanModeType mode = torch::kFanIn,
            |     ~~~~~~~~~~~~^~~~~~~~~~~~~~~~~~~~
      /usr/local/lib/python3.12/dist-packages/torch/include/torch/csrc/api/include/torch/nn/modules/conv.h: In instantiation of ‘void torch::nn::ConvNdImpl<D, Derived>::reset_parameters() [with long unsigned int D = 2; Derived = torch::nn::Conv2dImpl]’:
      /usr/local/lib/python3.12/dist-packages/torch/include/torch/csrc/api/include/torch/nn/modules/conv.h:100:5:   required from ‘void torch::nn::ConvNdImpl<D, Derived>::reset() [with long unsigned int D = 2; Derived = torch::nn::Conv2dImpl]’
      /usr/local/lib/python3.12/dist-packages/torch/include/torch/csrc/api/include/torch/nn/modules/conv.h:32:8:   required from here
      /usr/local/lib/python3.12/dist-packages/torch/include/torch/csrc/api/include/torch/nn/modules/conv.h:104:27: error: cannot convert ‘const torch::enumtype::kFanIn’ to ‘int’
        104 |     init::kaiming_uniform_(
            |     ~~~~~~~~~~~~~~~~~~~~~~^
            |                           |
            |                           const torch::enumtype::kFanIn
        105 |         weight,
            |         ~~~~~~~
        106 |         /*a=*/std::sqrt(5)); // NOLINT(cppcoreguidelines-avoid-magic-numbers)
            |         ~~~~~~~~~~~~~~~~~~~
      /usr/local/lib/python3.12/dist-packages/torch/include/torch/csrc/api/include/torch/nn/init.h:98:17: note:   initializing argument 3 of ‘at::Tensor torch::nn::init::kaiming_uniform_(at::Tensor, double, int, int)’
         98 |     FanModeType mode = torch::kFanIn,
            |     ~~~~~~~~~~~~^~~~~~~~~~~~~~~~~~~~
      /usr/local/lib/python3.12/dist-packages/torch/include/torch/csrc/api/include/torch/nn/modules/conv.h: In instantiation of ‘void torch::nn::ConvNdImpl<D, Derived>::reset_parameters() [with long unsigned int D = 1; Derived = torch::nn::Conv1dImpl]’:
      /usr/local/lib/python3.12/dist-packages/torch/include/torch/csrc/api/include/torch/nn/modules/conv.h:100:5:   required from ‘void torch::nn::ConvNdImpl<D, Derived>::reset() [with long unsigned int D = 1; Derived = torch::nn::Conv1dImpl]’
      /usr/local/lib/python3.12/dist-packages/torch/include/torch/csrc/api/include/torch/nn/modules/conv.h:32:8:   required from here
      /usr/local/lib/python3.12/dist-packages/torch/include/torch/csrc/api/include/torch/nn/modules/conv.h:104:27: error: cannot convert ‘const torch::enumtype::kFanIn’ to ‘int’
        104 |     init::kaiming_uniform_(
            |     ~~~~~~~~~~~~~~~~~~~~~~^
            |                           |
            |                           const torch::enumtype::kFanIn
        105 |         weight,
            |         ~~~~~~~
        106 |         /*a=*/std::sqrt(5)); // NOLINT(cppcoreguidelines-avoid-magic-numbers)
            |         ~~~~~~~~~~~~~~~~~~~
      /usr/local/lib/python3.12/dist-packages/torch/include/torch/csrc/api/include/torch/nn/init.h:98:17: note:   initializing argument 3 of ‘at::Tensor torch::nn::init::kaiming_uniform_(at::Tensor, double, int, int)’
         98 |     FanModeType mode = torch::kFanIn,
            |     ~~~~~~~~~~~~^~~~~~~~~~~~~~~~~~~~
      error: command '/usr/bin/x86_64-linux-gnu-gcc' failed with exit code 1
      [end of output]
  
  note: This error originates from a subprocess, and is likely not a problem with pip.
  ERROR: Failed building wheel for patrickstar
  Running setup.py clean for patrickstar
Failed to build patrickstar
ERROR: Could not build wheels for patrickstar, which is required to install pyproject.toml-based projects

ubuntu 24.04 cuda 13.0 pytorch 2.11