search

yoyo-nb / Thin-Plate-Spline-Motion-Model

[CVPR 2022] Thin-Plate Spline Motion Model for Image Animation.
MIT License
3.39k stars 555 forks source link

CUDA_VISIBLE_DEVICES can not be found or is misspelled #34

Open eizoxx opened 1 year ago

eizoxx commented 1 year ago

Hey, I am now on the final step of the installation and just need to enter the command: _CUDA_VISIBLE_DEVICES=0 python demo.py --config config/vox-256.yaml --checkpoint checkpoints/vox.pth.tar --source_image ./source.jpg --drivingvideo ./driving.mp4

But when I paste it in the command line, it tells me: "The command "CUDA_VISIBLE_DEVICES" is either misspelled or could not be found".

What did I do wrong?

unknownentity123 commented 1 year ago

I always use "set CUDA_VISIBLE_DEVICES=1 & python......" when using local repos in an Anaconda env....the usage of the "&" was recommended on another repo and since using that I haven't had any issues

Where I use 1 for my second 1070ti card instead of my first card

eizoxx commented 1 year ago

Now is says:

Original: Der Befehl "CUDA_VISIBLE_DEVICES" ist entweder falsch geschrieben oder konnte nicht gefunden werden. Traceback (most recent call last): File "D:\AI\Thin-Plate-Spline-Motion-Model\demo.py", line 14, in from modules.keypoint_detector import KPDetector File "D:\AI\Thin-Plate-Spline-Motion-Model\modules\keypoint_detector.py", line 3, in from torchvision import models ModuleNotFoundError: No module named 'torchvision'

English: The command "CUDA_VISIBLE_DEVICES" is either misspelled or could not be found Traceback (most recent call last): File "D:\AI\Thin-Plate-Spline-Motion-Model\demo.py", line 14, in from modules.keypoint_detector import KPDetector File "D:\AI\Thin-Plate-Spline-Motion-Model\modules\keypoint_detector.py", line 3, in from torchvision import models ModuleNotFoundError: No module named 'torchvision'

However, the first line is in German. But actually this should not make a difference, because all other commands had also worked

yoyo-nb commented 1 year ago

You got the error: ModuleNotFoundError: No module named 'torchvision' You can follow the instructions on pytorch.org to install the appropriate version of torch and torchvision

spider853 commented 1 year ago

General advices for Linux commands on Windows: When you see something like XXX=NUMBER python ... it doesn't work on Windows cmd to make it work, separate the first part before python and set the enviornament variables manually set CUDA_VISIBLE_DEVICES=0 and hit enter then enter the second part from and including pyton ...