Nvidia GeForce 30xx / Cuda 11

Qiulin-W / SAFA

Official Pytorch Implementation of 3DV2021 paper: SAFA: Structure Aware Face Animation.

Other

177 stars 29 forks source link

Nvidia GeForce 30xx / Cuda 11 #5

Open Lorjuo opened 2 years ago

Lorjuo commented 2 years ago

Did maybe somebody got SAFA running with a Nvidia GeForce 30xx? As far as I know this series requires Cuda 11.x. Currently I experienced a variety of different errors when trying to install this module.

If you had some success, could you please post the output of "pip list" or provide some instructions / hints?

Thanks

gsotnikov commented 2 years ago

@Lorjuo Hi! I did it with 3090, my CUDA set up is: Cuda compilation tools, release 11.5, Ubuntu

Use pytorch 1.10, CUDA 11.3
Core trick, which worked for me is installing pytorch3d via pip install --no-index --no-cache-dir pytorch3d -f https://dl.fbaipublicfiles.com/pytorch3d/packaging/wheels/py39_cu113_pyt1100/download.html