Torch-encoding for cuda 11.6

torch.package / torch::deploy
1

I am struggling to install the (torch-encoding) package for cuda 11.6

I have no problem using torch itself, but i can’t build or install torch-encoding with pip or conda.

error:

ld/temp.linux-x86_64-cpython-310/tmp/pip-install-t74d_udw/torch-encoding_3d239eb112774b44aa6d43ce06ef2dfe/encoding/lib/gpu/activation_kernel.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'"'"'' -DCUDA_HAS_FP16=1 -D__CUDA_NO_HALF_OPERATORS__ -D__CUDA_NO_HALF_CONVERSIONS__ -D__CUDA_NO_HALF2_OPERATORS__ -DTORCH_API_INCLUDE_EXTENSION_H '-DPYBIND11_COMPILER_TYPE="_gcc"' '-DPYBIND11_STDLIB="_libstdcpp"' '-DPYBIND11_BUILD_ABI="_cxxabi1011"' -DTORCH_EXTENSION_NAME=gpu -D_GLIBCXX_USE_CXX11_ABI=0 -gencode=arch=compute_75,code=compute_75 -gencode=arch=compute_75,code=sm_75 -std=c++14
      ninja: build stopped: subcommand failed.
      Traceback (most recent call last):
        File "/home/engrapher/anaconda3/envs/langseg2/lib/python3.10/site-packages/torch/utils/cpp_extension.py", line 1900, in _run_ninja_build
          subprocess.run(
        File "/home/engrapher/anaconda3/envs/langseg2/lib/python3.10/subprocess.py", line 526, in run
          raise CalledProcessError(retcode, process.args,
      subprocess.CalledProcessError: Command '['ninja', '-v']' returned non-zero exit status 1.

Any help is appreciated

2

Based on the install instructions it seems this lib depends on torch==1.4.0, so I doubt it was updated and compatible with the latest PyTorch or CUDA release. You could post an issue in their GitHub repository and ask if this package is still maintained.

3

Looks like I should have done some research myself.

Thanks a lot!