ZiqiaoPeng / SyncTalk

[CVPR 2024] This is the official source for our paper "SyncTalk: The Devil is in the Synchronization for Talking Head Synthesis"
https://ziqiaopeng.github.io/synctalk/
Other
1.07k stars 119 forks source link

Does it work with different CUDA and pytorch versions? #181

Open kamadforge opened 1 week ago

kamadforge commented 1 week ago

Did it work for you with any other version than pip install torch==1.12.1+cu113 torchvision==0.13.1+cu113 torchaudio==0.12.1 --extra-index-url https://download.pytorch.org/whl/cu113 ?

I have cuda 12.4 on Ubuntu 22.04 and when I installed

conda install pytorch torchvision torchaudio pytorch-cuda=12.4 -c pytorch-nightly -c nvidia

I have issues installing freqencoder, shencoder, etc. I'm not sure if the encoders cannot compile because of this different cuda versions but perhaps anyone has experienced compiling them with a different cuda version

hungho77 commented 1 week ago

install cuda 11.3 for pytorch 3d, after that you install newest cuda and pytorch version for remaining install. I have tried, and it work ok

kamadforge commented 1 week ago

Interestingly, pytorch 3d ran fine with this set-up but the packages produce an error: ERROR: Failed building wheel for freqencoder etc.

pip install ./freqencoder pip install ./shencoder pip install ./gridencoder pip install ./raymarching

Are thete any particular CUDA/pytorch requirments for these packages?

hungho77 commented 1 week ago

yes, after install the pytorch3d, you have to reinstall pytorch match with your cuda toolkit version, then install above package