在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就應該沒有問題了。