Yu
Results
2
comments of
Yu
I got same issue with ubuntu 20.04, python 3.7, PyTorch 1.5 and torchvision 0.6. Solved by run`export TORCH_CUDA_ARCH_LIST="7.0"` before `python src/trackformer/models/ops/setup.py build --build-base=src/trackformer/models/ops/ install`.
1. Run the command `nvidia-smi ` to identify GPU architecture NVIDIA A10G, look up Compute Capability: 8.6 2. Run `nvcc --version` to check CUDA toolkit version: 11.3, which compatible with...