Unknown CUDA arch (8.6) or GPU not supported

timmeinhardt / trackformer

Implementation of "TrackFormer: Multi-Object Tracking with Transformers”. [Conference on Computer Vision and Pattern Recognition (CVPR), 2022]

https://arxiv.org/abs/2101.02702

Apache License 2.0

487 stars 113 forks source link

Unknown CUDA arch (8.6) or GPU not supported #117

Open rumjie opened 6 months ago

rumjie commented 6 months ago

Hi, hope you are well.

I am trying to use trackformer with docker, and I tried several environments and all failed. I guess using image of pytorch makes error with installing lapsolver (or wheel, I guess). And ubuntu:20.04 and nvidia/cuda makes same error like title - Unknown CUDA arch (8.6) or GPU not supported . Could you recommend solutions or other versions of python, torch or cuda for dependency of trackformer?

Ricardo-Yu commented 4 months ago

I got same issue with ubuntu 20.04, python 3.7, PyTorch 1.5 and torchvision 0.6. Solved by runexport TORCH_CUDA_ARCH_LIST="7.0" before python src/trackformer/models/ops/setup.py build --build-base=src/trackformer/models/ops/ install.