nvcc fatal : Unsupported gpu architecture 'compute_86'

[Tungthanhlee]

Thank you for publishing a great repo.
I am building the environment for the repo and got this error after running git checkout tags/vox2cortex_cvpr2022 -b vox2cortex_pytorch3d pip install -e . Im running on CUDA 11.7 on NVIDIA A6000; Python 3.8
Could you address it?

`Running setup.py develop for pytorch3d
error: subprocess-exited-with-error

× python setup.py develop did not run successfully.
│ exit code: 1
╰─> [53 lines of output]
    running develop
    running egg_info
    creating pytorch3d.egg-info
    writing pytorch3d.egg-info/PKG-INFO
    writing dependency_links to pytorch3d.egg-info/dependency_links.txt
    writing requirements to pytorch3d.egg-info/requires.txt
    writing top-level names to pytorch3d.egg-info/top_level.txt
    writing manifest file 'pytorch3d.egg-info/SOURCES.txt'
    reading manifest file 'pytorch3d.egg-info/SOURCES.txt'
    adding license file 'LICENSE'
    writing manifest file 'pytorch3d.egg-info/SOURCES.txt'
    running build_ext
    building 'pytorch3d._C' extension
    creating build
    creating build/temp.linux-x86_64-cpython-38
    creating build/temp.linux-x86_64-cpython-38/home
    creating build/temp.linux-x86_64-cpython-38/home/thanhtul
    creating build/temp.linux-x86_64-cpython-38/home/thanhtul/code
    creating build/temp.linux-x86_64-cpython-38/home/thanhtul/code/pytorch3d
    creating build/temp.linux-x86_64-cpython-38/home/thanhtul/code/pytorch3d/pytorch3d
    creating build/temp.linux-x86_64-cpython-38/home/thanhtul/code/pytorch3d/pytorch3d/csrc
    creating build/temp.linux-x86_64-cpython-38/home/thanhtul/code/pytorch3d/pytorch3d/csrc/blending
    creating build/temp.linux-x86_64-cpython-38/home/thanhtul/code/pytorch3d/pytorch3d/csrc/compositing
    creating build/temp.linux-x86_64-cpython-38/home/thanhtul/code/pytorch3d/pytorch3d/csrc/face_areas_normals
    creating build/temp.linux-x86_64-cpython-38/home/thanhtul/code/pytorch3d/pytorch3d/csrc/gather_scatter
    creating build/temp.linux-x86_64-cpython-38/home/thanhtul/code/pytorch3d/pytorch3d/csrc/interp_face_attrs
    creating build/temp.linux-x86_64-cpython-38/home/thanhtul/code/pytorch3d/pytorch3d/csrc/knn
    creating build/temp.linux-x86_64-cpython-38/home/thanhtul/code/pytorch3d/pytorch3d/csrc/mesh_normal_consistency
    creating build/temp.linux-x86_64-cpython-38/home/thanhtul/code/pytorch3d/pytorch3d/csrc/packed_to_padded_tensor
    creating build/temp.linux-x86_64-cpython-38/home/thanhtul/code/pytorch3d/pytorch3d/csrc/point_mesh
    creating build/temp.linux-x86_64-cpython-38/home/thanhtul/code/pytorch3d/pytorch3d/csrc/pulsar
    creating build/temp.linux-x86_64-cpython-38/home/thanhtul/code/pytorch3d/pytorch3d/csrc/pulsar/cuda
    creating build/temp.linux-x86_64-cpython-38/home/thanhtul/code/pytorch3d/pytorch3d/csrc/pulsar/host
    creating build/temp.linux-x86_64-cpython-38/home/thanhtul/code/pytorch3d/pytorch3d/csrc/pulsar/pytorch
    creating build/temp.linux-x86_64-cpython-38/home/thanhtul/code/pytorch3d/pytorch3d/csrc/rasterize_meshes
    creating build/temp.linux-x86_64-cpython-38/home/thanhtul/code/pytorch3d/pytorch3d/csrc/rasterize_points
    /usr/bin/nvcc -DWITH_CUDA -DTHRUST_IGNORE_CUB_VERSION_CHECK -I/home/thanhtul/code/pytorch3d/pytorch3d/csrc -I/home/thanhtul/miniconda3/envs/py38/lib/python3.8/site-packages/torch/include -I/home/thanhtul/miniconda3/envs/py38/lib/python3.8/site-packages/torch/include/torch/csrc/api/include -I/home/thanhtul/miniconda3/envs/py38/lib/python3.8/site-packages/torch/include/TH -I/home/thanhtul/miniconda3/envs/py38/lib/python3.8/site-packages/torch/include/THC -I/home/thanhtul/miniconda3/envs/py38/include/python3.8 -c /home/thanhtul/code/pytorch3d/pytorch3d/csrc/blending/sigmoid_alpha_blend.cu -o build/temp.linux-x86_64-cpython-38/home/thanhtul/code/pytorch3d/pytorch3d/csrc/blending/sigmoid_alpha_blend.o -D__CUDA_NO_HALF_OPERATORS__ -D__CUDA_NO_HALF_CONVERSIONS__ -D__CUDA_NO_HALF2_OPERATORS__ --expt-relaxed-constexpr --compiler-options '-fPIC' -std=c++14 -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=_C -D_GLIBCXX_USE_CXX11_ABI=0 -gencode=arch=compute_86,code=sm_86
    nvcc fatal   : Unsupported gpu architecture 'compute_86'
    /home/thanhtul/code/pytorch3d/setup.py:72: UserWarning: The environment variable `CUB_HOME` was not found. NVIDIA CUB is required for compilation and can be downloaded from `https://github.com/NVIDIA/cub/releases`. You can unpack it to a location of your choice and set the environment variable `CUB_HOME` to the folder containing the `CMakeListst.txt` file.
      warnings.warn(
    /home/thanhtul/miniconda3/envs/py38/lib/python3.8/site-packages/setuptools/command/easy_install.py:144: EasyInstallDeprecationWarning: easy_install command is deprecated. Use build and pip and other standards-based tools.
      warnings.warn(
    /home/thanhtul/miniconda3/envs/py38/lib/python3.8/site-packages/setuptools/command/install.py:34: SetuptoolsDeprecationWarning: setup.py install is deprecated. Use build and pip and other standards-based tools.
      warnings.warn(
    /home/thanhtul/miniconda3/envs/py38/lib/python3.8/site-packages/torch/utils/cpp_extension.py:352: UserWarning: Attempted to use ninja as the BuildExtension backend but we could not find ninja.. Falling back to using the slow distutils backend.
      warnings.warn(msg.format('we could not find ninja.'))
    /home/thanhtul/miniconda3/envs/py38/lib/python3.8/site-packages/torch/cuda/__init__.py:104: UserWarning:
    NVIDIA RTX A6000 with CUDA capability sm_86 is not compatible with the current PyTorch installation.
    The current PyTorch install supports CUDA capabilities sm_37 sm_50 sm_60 sm_70 sm_75.
    If you want to use the NVIDIA RTX A6000 GPU with PyTorch, please check the instructions at https://pytorch.org/get-started/locally/

      warnings.warn(incompatible_device_warn.format(device_name, capability, " ".join(arch_list), device_name))
    error: command '/usr/bin/nvcc' failed with exit code 1
    [end of output]

note: This error originates from a subprocess, and is likely not a problem with pip.`