first-order-model
first-order-model copied to clipboard
Running on remote CentOS 7 Server
I'm trying to build a closed web API where users can upload a pic of themselves and I'll make them sing a song like WOMBO does. Anyways, when I try to run it, it fails at the driver:
When I run:
docker run -it --rm --gpus all \
-v $HOME/first-order-model:/app first-order-model \
python3 demo.py \
--config config/vox-256.yaml \
--driving_video input/iko-iko.mp4 \
--source_image input/iko-iko.jpg \
--checkpoint checkpoints/vox-cpk.pth.tar \
--result_video output/iko-iko.mp4 \
--relative --adapt_scale
It gives me this error:
docker: Error response from daemon: could not select device driver "" with capabilities: [[gpu]].
I'm assuming it has to do with nvidia-docker, but i have no clue. Will this project only work on desktop?
Thanks! Joe