Не удалось инициализировать gpu для этого экземпляра. ОШИБКА В FLOYDHUB CLI: «Поддержка GPU отключена. Скомпилируйте MXNet с USE_CUDA = 1, чтобы включить поддержку GPU». - PullRequest
Новая
       46

Не удалось инициализировать gpu для этого экземпляра. ОШИБКА В FLOYDHUB CLI: «Поддержка GPU отключена. Скомпилируйте MXNet с USE_CUDA = 1, чтобы включить поддержку GPU».

0 голосов
/ 01 ноября 2019

Я пытаюсь обучить мою модель обнаружения объектов MXNet YOLO на FloydHub, но я не могу инициализировать GPU для этого экземпляра.

ОШИБКА В CLI FLOYDHUB: «Поддержка GPU отключена. Скомпилируйте MXNet с USE_CUDA = 1, чтобы включить поддержку GPU.»

Я использовал pip для установки mxnet-cu92mkl /cu92 / CU100. Попытка загрузить мою работу с другой версией CUDA, но каждый раз, когда я получаю ту же ошибку. Я использую облачную платформу FloydHub. 

# training contexts
try:
    a = mx.nd.zeros((1,), ctx=mx.gpu(0))
    ctx = [mx.gpu(0)]
except:
    ctx = [mx.cpu()]
print(f"Training model on {ctx[0].device_type.upper()} ...")
print("Cpu or Gpu is decided decision=",ctx[0])
...