Giters

ThibaultGROUEIX / ChamferDistancePytorch

Chamfer Distance in Pytorch with f-score

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

How to build with CUDA11+?

ark1234 opened this issue · comments

commented

Thank you so much for your good work!
I have a A100 GPU, it only works for cuda11+ ,how to compile it in cuda11+. I tried to add "export TORCH_CUDA_ARCH_LIST="7.5"" on bashrc, it can compile, but did not work

commented

figure it out

commented

figure it out

Hi bro, how do you compile it in cuda11+?

commented

figure it out

Hi bro, how do you compile it in cuda11+?

give an example, my cuda vision is 11.3 and then install pytorch use "conda install pytorch==1.11.0 torchvision==0.12.0 torchaudio==0.11.0 cudatoolkit=11.3 -c pytorch",it maybe work.