Cuda Arg [+] index out of range and CUD_HOME not detected

[NicCusinato]

File "/usr/local/lib/python3.8/dist-packages/torch/utils/cpp_extension.py", line 1773, in _get_cuda_arch_flags
31.59 arch_list[-1] += '+PTX'
31.59 IndexError: list index out of range
31.59 ----------------------------------------
31.64 ERROR: Command errored out with exit status 1: /usr/bin/python3 -u -c 'import sys, setuptools, tokenize; sys.argv[0] = '"'"'/tmp/pip-req-build-4pbnudhf/setup.py'"'"'; file='"'"'/tmp/pip-req-build-4pbnudhf/setup.py'"'"';f=getattr(tokenize, '"'"'open'"'"', open)(file);code=f.read().replace('"'"'\r\n'"'"', '"'"'\n'"'"');f.close();exec(compile(code, file, '"'"'exec'"'"'))' install --record /tmp/pip-record-65ips369/install-record.txt --single-version-externally-managed --compile --install-headers /usr/local/include/python3.8/pytorch3d Check the logs for full command output.

Building wheel for pytorch3d (setup.py): finished with status 'error'
24.95 ERROR: Command errored out with exit status 1:
24.95 command: /usr/bin/python3 -u -c 'import sys, setuptools, tokenize; sys.argv[0] = '"'"'/tmp/pip-req-build-4pbnudhf/setup.py'"'"'; file='"'"'/tmp/pip-req-build-4pbnudhf/setup.py'"'"';f=getattr(tokenize, '"'"'open'"'"', open)(file);code=f.read().replace('"'"'\r\n'"'"', '"'"'\n'"'"');f.close();exec(compile(code, file, '"'"'exec'"'"'))' bdist_wheel -d /tmp/pip-wheel-ojl3fmb4
24.95 cwd: /tmp/pip-req-build-4pbnudhf/
24.95 Complete output (321 lines):
24.95 No CUDA runtime is found, using CUDA_HOME='/usr/local/cuda'
24.95 /tmp/pip-req-build-4pbnudhf/setup.py:84: 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.
24.95 warnings.warn(