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

Author: aestendrelaCreated Jan 31, 2023Updated Oct 17, 2023

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?

Source: NVIDIA/vid2vid