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

[aestendrela]

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]

@aestendrela , are you able to get the solution?