search

ThibaultGROUEIX / ChamferDistancePytorch

Chamfer Distance in Pytorch with f-score
MIT License
326 stars 43 forks source link

How to build with CUDA11+? #29

Closed ark1234 closed 1 year ago

ark1234 commented 1 year ago

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

ark1234 commented 1 year ago

figure it out

kanonjz commented 1 year ago

figure it out

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

HanwenWangBUPT commented 1 year ago

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.