Is this project compatible with a recent GPU ?
antoinersw opened this issue · 9 comments
I have a RTX 4080 (sm__89). It doesnt seem to be compatible with pytorch 1.6
https://pytorch.org/get-started/locally/
Do you believe I can use the nightly build with a more recent pytorch3d version ?
Thanks
It works with my 3090 on Windows and WSL Ubuntu. Off the top of my head, you need to use a more recent version of PyTorch.
@emlcpfx can you help me im getting errors in running the code
I’m a bit rusty, I barely remember this, but I’ll try. What are the errors?
This is a guide to getting EMOCA to work on Windows. EMOCA is built on top of DECA, so maybe it’ll help— https://docs.google.com/document/d/13rbNHEtyFqbvgHoIrymD558wmTgd7A67zPE85ISe144/edit
im running the project on google colab
@emlcpfx can i get your email id or something else where i can contact you
Leave yours, and I’ll email you.
@zardamhussain how did you get this working on colab? the notebook provided is not compatible and i could not install any of the requirements
-
download the miniconda in google-colab by using these command
! wget https://repo.anaconda.com/miniconda/Miniconda3-py37_4.8.2-Linux-x86_64.sh
! chmod +x Miniconda3-py38_4.8.3-Linux-x86_64.sh
! bash ./Miniconda3-py38_4.8.3-Linux-x86_64.sh -b -f -p /usr/local
import sys
sys.path.append('/usr/local/lib/python3.8/site-packages/')
-
then run these comamands
!conda install pytorch==1.7.1 torchvision==0.8.2 torchaudio==0.7.2 cudatoolkit=11.0 -c pytorch
!conda install -c fvcore -c iopath -c conda-forge fvcore iopath
!conda install pytorch3d -c pytorch3d
-
pip install -r requirements.txt
numpy
scipy
chumpy
scikit-image
opencv-python
scikit-image #skimage
PyYAML
# torch==1.6.0 # for compatible with pytorch3d
# torchvision==0.7.0
face-alignment
yacs
kornia==0.5.0
ninja
# fvcore
# pytorch3d