Provide instructions for GPU based pytorch

PeizhuoLi / neural-blend-shapes

An end-to-end library for automatic character rigging, skinning, and blend shapes generation, as well as a visualization tool [SIGGRAPH 2021]

Other

637 stars 92 forks source link

Closed fire closed 3 years ago

fire commented 3 years ago

Similar to the current cpu version. Provide an environment.yml for cuda based pytorch.

It is non trivial to switch between gpu and cpu. https://github.com/pytorch/pytorch/issues/30664.

fire commented 3 years ago

The way to do it to create a new environment and run something similar to:

conda install tensorboard tqdm install pytorch torchvision torchaudio cudatoolkit=11.1 -c pytorch -c nvidia

Mamba solves dependencies faster than conda.