MXNetError: [15:25:07] C: \ Jenkins \ workspace \ mxnet-tag \ mxnet \ src \ storage \ storage.cc: 137: скомпилировать с USE_CUDA = 1, чтобы включить использование графического процессора - PullRequest
Новая
       9

MXNetError: [15:25:07] C: \ Jenkins \ workspace \ mxnet-tag \ mxnet \ src \ storage \ storage.cc: 137: скомпилировать с USE_CUDA = 1, чтобы включить использование графического процессора

0 голосов
/ 11 января 2019

Я пытался обучить свой код с GPU в Jupyter. Но я не понимаю правильный GPU.

моя среда: Windows 10, cuda8, python3.6, mxnetgpu, Юпитер.

мой код: 

a = nd.array([1, 2, 3], ctx=mx.gpu())

Но моя ошибка такая: 

MXNetError                                Traceback (most recent call last)
    <ipython-input-6-3c78e5d2ccff> in <module>
    ----> 1 a = nd.array([1, 2, 3], ctx=mx.gpu())
          2 a

    d:\data\python\lib\site-packages\mxnet\ndarray\utils.py in array(source_array, ctx, dtype)
    144         return _sparse_array(source_array, ctx=ctx, dtype=dtype)
    145     else:
--> 146         return _array(source_array, ctx=ctx, dtype=dtype)
    147 
    148 

d:\data\python\lib\site-packages\mxnet\ndarray\ndarray.py in array(source_array, ctx, dtype)
   2432             except:
   2433                 raise TypeError('source_array must be array like object')
-> 2434     arr = empty(source_array.shape, ctx, dtype)
   2435     arr[:] = source_array
   2436     return arr

d:\data\python\lib\site-packages\mxnet\ndarray\ndarray.py in empty(shape, ctx, dtype)
   3818     if dtype is None:
   3819         dtype = mx_real_t
-> 3820     return NDArray(handle=_new_alloc_handle(shape, ctx, False, dtype))
   3821 
   3822 

d:\data\python\lib\site-packages\mxnet\ndarray\ndarray.py in _new_alloc_handle(shape, ctx, delay_alloc, dtype)
    137         ctypes.c_int(int(delay_alloc)),
    138         ctypes.c_int(int(_DTYPE_NP_TO_MX[np.dtype(dtype).type])),
--> 139         ctypes.byref(hdl)))
    140     return hdl
    141 

d:\data\python\lib\site-packages\mxnet\base.py in check_call(ret)
    250     """
    251     if ret != 0:
--> 252         raise MXNetError(py_str(_LIB.MXGetLastError()))
    253 
    254 

MXNetError: [15:25:07] C:\Jenkins\workspace\mxnet-tag\mxnet\src\storage\storage.cc:137: Compile with USE_CUDA=1 to enable GPU usage

Я изменяю свой путь добавления файла конфигурации и ИСПОЛЬЗУЮ CUDA. Но он не работает! Как его решить?

1 Ответ

0 голосов
/ 16 января 2019

Похоже, вы установили сборку MXNet-only CPU в виде pip-пакета. Вам нужно будет удалить сборку процессора

pip uninstall mxnet

и установите сборку mxnet с поддержкой cuda для вашей версии cuda (8.0)

pip install mxnet-cu80

См. http://mxnet.incubator.apache.org/install/index.html?platform=Linux&language=Python&processor=GPU.

Если вы получаете ошибки с жалобами на CUDA, вам, возможно, придется следовать инструкциям по настройке CUDA_PATH: https://docs.nvidia.com/cuda/cuda-installation-guide-microsoft-windows/

Vishaal

...