GCP DL Image: База m36 (с CUDA 10.0). Как включить графический процессор в PyTorch? - PullRequest
1 голос
/ 06 ноября 2019

Я пытаюсь запустить задания DL PyTorch на блоке GCP. Шаги, которые я предпринимаю:

  • Создайте виртуальную машину с образом глубокого обучения m36 и приложенным K80. При первом входе в систему возникает вопрос об установке драйвера CUDA 10, я выбираю 'y', и драйвер настраивается.
  • install pyenv
  • pyenv install 3.7.5
  • pyenv virtualenv 3.7.5 myenv
  • pyenv activate myenv
  • pip install fastai
  • Запустить консоль Python и запустить:

Код:

import torch
torch.cuda.is_available()

Возвращает False. Что я должен сделать, чтобы включить графический процессор в PyTorch?

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...