@Koteyka1224

Как исправить ошибку в Python & torch?

Всем добрый день!
Решил я написать нейронную сеть на python. За основу взял вот этот проект (https://github.com/junyanz/pytorch-CycleGAN-and-pix2pix). Перед тем, как писать свой код, решил потестить образец. При запуске кода возникает такая ошибка:
Traceback (most recent call last):
  File "D:\Home\Programming\Python\SG\pytorch-CycleGAN-and-pix2pix-master\pytorch-CycleGAN-and-pix2pix-master\train.py", line 28, in <module>
    opt = TrainOptions().parse()   # get training options
  File "D:\Home\Programming\Python\SG\pytorch-CycleGAN-and-pix2pix-master\pytorch-CycleGAN-and-pix2pix-master\options\base_options.py", line 133, in parse
    torch.cuda.set_device(opt.gpu_ids[0])
  File "C:\Users\KorolevEY\AppData\Roaming\Python\Python37\site-packages\torch\cuda\__init__.py", line 264, in set_device
    torch._C._cuda_setDevice(device)
AttributeError: module 'torch._C' has no attribute '_cuda_setDevice'


Уже третий день не могу пофиксить эту проблему. Любой помощи буду благодарен. Заранее спасибо :)
  • Вопрос задан
  • 124 просмотра
Пригласить эксперта
Ваш ответ на вопрос

Войдите, чтобы написать ответ

Войти через центр авторизации
Похожие вопросы