Thin-Plate-Spline-Motion-Model icon indicating copy to clipboard operation
Thin-Plate-Spline-Motion-Model copied to clipboard

CUDA_VISIBLE_DEVICES can not be found or is misspelled

Open eizoxx opened this issue 2 years ago • 4 comments

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 --driving_video ./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?

eizoxx avatar Oct 01 '22 16:10 eizoxx

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

unknownentity123 avatar Oct 01 '22 16:10 unknownentity123

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

eizoxx avatar Oct 02 '22 13:10 eizoxx

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

yoyo-nb avatar Oct 11 '22 02:10 yoyo-nb

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 ...

spider853 avatar Dec 02 '22 15:12 spider853