Python перестал работать над переключением в контекст GPU

Question

Я пытаюсь настроить свою среду машинного обучения в AWS следующим образом: -

OS: windows server 2012 r2 , 64 bit
instance: p2.xlarge
GPU : Tesla K80 series
CUDA: 9.2.148
Graphis driver: 398.26 (installed by cuda toolkit)
python : 3.5 ( tested using 2.7.15 as well) ,64 bit
IDE: Pycharm Community 2018.2, 64 bit
mxnet librabry: mxnet-cu92

теперь, когда я запускаю следующий код:

import mxnet as mx
from mxnet import nd
mx.random.seed(1)
z = nd.ones(shape=(3,3), ctx=mx.cpu())
print(z)

, он работает нормально, но когда яизменить ctx = mx.gpu () или ctx = mx.gpu (0) , я получаю сообщение об ошибке Python перестал работать.

Настройка CUDA работает нормально, я скомпилировал и запустил deviceQuery,bandwidthTest пример приложения, они выдают результат, как и ожидалось.

EDIT :: подробности сбоя python дают мне следующую информацию:

Fault Module Name:  ucrtbase.DLL

Я пробовал с драйвером Cuda Graphis: 398.44 (рекомендуется изсайт cuda для моего gpu и os) до сих пор не повезло.

Есть идеи по решению этой проблемы?

Answer 1

Наконец-то я получил решение, потратив почти 3 дня, это была библиотека mxnet, которая вызывала проблемы. После того, как я попробовал разные библиотеки cuda toolkit и mxnet, у меня сработала следующая комбинация: -

1) Cuda Toolkit - 9.2
   Mxnet lbrary : mxnet-cu92 version 1.2.0b20180525
   (Failing for most of the recent versions ) 
2)  Cuda Toolkit - 8.0
   Mxnet lbrary : mxnet-cu80 version 0.11.0
   (Failing for most of the recent versions  ) 

спасибо