terminate called after throwing an instance of dmlc::Error

在python中使用MxNet時在代碼

 self.module.bind(data_shapes=[data_name_shape],
                         for_training=False)

報出一下錯誤

terminate called after throwing an instance of 'dmlc::Error'
  what():  [07:08:37] src/storage/storage.cc:119: Compile with USE_CUDA=1 to enable GPU usage
Stack trace:

主要原因是使用

pip install mxnet

默認安裝得不是gpu版得,應該安裝對應cuda版本得mxnet就應該沒有問題了。

cuda10.1得下載地址

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章