Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Error installing mmdet3d, IndexError out of range #364

Open
QingyangLiu opened this issue Sep 9, 2024 · 1 comment
Open

Error installing mmdet3d, IndexError out of range #364

QingyangLiu opened this issue Sep 9, 2024 · 1 comment

Comments

@QingyangLiu
Copy link

After I built the Docker image and ran the docker container, I git pulled the repo and ran "pip install -v -e ." However, I then got the following error message when it was trying to install mmdet3d:

File "/opt/conda/lib/python3.8/site-packages/torch/utils/cpp_extension.py", line 1606, in _get_cuda_arch_flags
arch_list[-1] += '+PTX'
IndexError: list index out of range
error: subprocess-exited-with-error

I'm confused as to how to proceed. How should this be fixed? I attached the full error message.
Screenshot from 2024-09-09 19-08-29

@brankyy
Copy link

brankyy commented Nov 7, 2024

export TORCH_CUDA_ARCH_LIST="8.6+PTX"
works for me with RTX3090.
try export TORCH_CUDA_ARCH_LIST="YOUR_GPU_CC+PTX"

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants