NVIDIA / vid2vid

Pytorch implementation of our method for high-resolution (e.g. 2048x1024) photorealistic video-to-video translation.
Other
8.57k stars 1.2k forks source link

AttributeError: module 'torch._C' has no attribute '_cuda_setDevice' #175

Open aestendrela opened 1 year ago

aestendrela commented 1 year ago

When I run this code it gives me an error

python test.py --name edge2face_512 --dataroot datasets/face/ --dataset_mode face --input_nc 15 --loadSize 512 --use_single_G

ERROR: AttributeError: module 'torch._C' has no attribute '_cuda_setDevice'.

Do you know what it could be?

hacker009-sudo commented 10 months ago

@aestendrela , are you able to get the solution?