trackformer icon indicating copy to clipboard operation
trackformer copied to clipboard
verified publisher icon timmeinhardt

Unknown CUDA arch (8.6) or GPU not supported

Open rumjie opened this issue 2 years ago • 1 comments

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?

rumjie avatar Dec 27 '23 07:12 rumjie

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.

Ricardo-Yu avatar Feb 26 '24 18:02 Ricardo-Yu