Google Cloud - виртуальная машина Pytorch Deep Learning - ошибка: у объекта 'module' нет атрибута '_cuda_setDevice' - PullRequest
0 голосов
/ 03 февраля 2019

Я попробовал GCP, следуя этим инструкциям https://cloud.google.com/deep-learning-vm/docs/pytorch_start_instance, и я попытался запустить свой код, используя графический процессор TeslaK80.Это дает мне следующий AttributeError: у объекта 'module' нет атрибута '_cuda_setDevice'.За несколько дней до этого у меня была другая ВМ, и она работала нормально.Я создал еще одну виртуальную машину глубокого обучения, и теперь она дает мне эту ошибку.Я не понимаю, почему это происходит сейчас.

Это моя трассировка стека:

AttributeError                            Traceback (most recent call last)
<ipython-input-3-cc76841a4302> in <module>()
     45 # from config_training import config as config_training
     46 torch.manual_seed(0)
---> 47 torch.cuda.set_device(0)
     48 
     49 model = import_module(args.model)

/usr/local/lib/python2.7/dist-packages/torch/cuda/__init__.pyc in set_device(device)
    262     device = _get_device_index(device)
    263     if device >= 0:
--> 264         torch._C._cuda_setDevice(device)
    265 
    266 

AttributeError: 'module' object has no attribute '_cuda_setDevice'
Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...