NVIDIA / vid2vid

Pytorch implementation of our method for high-resolution (e.g. 2048x1024) photorealistic video-to-video translation.

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

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

aestendrela opened this issue · comments

commented

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?

@aestendrela , are you able to get the solution?