-
Notifications
You must be signed in to change notification settings - Fork 19
Description
I try GPU in Ubuntu server, but it appeared this error message
Error in mx.nd.internal.empty.array(shape, ctx) :
[05:09:27] src/storage/storage.cc:123: Compile with USE_CUDA=1 to enable GPU usage
Stack trace returned 10 entries:
[bt] (0) /usr/local/lib/R/site-library/mxnet/libs/libmxnet.so(dmlc::StackTraceabi:cxx11+0x4e) [0x7fd64cdd89ce]
[bt] (1) /usr/local/lib/R/site-library/mxnet/libs/libmxnet.so(dmlc::LogMessageFatal::~LogMessageFatal()+0x39) [0x7fd64cdd90f9]
[bt] (2) /usr/local/lib/R/site-library/mxnet/libs/libmxnet.so(+0x2ed69e7) [0x7fd64f8839e7]
[bt] (3) /usr/local/lib/R/site-library/mxnet/libs/libmxnet.so(+0x2ed8be5) [0x7fd64f885be5]
[bt] (4) /usr/loc

al/lib/R/site-library/mxnet/libs/libmxnet.so(mxnet::StorageImpl::Alloc(mxnet::Storage::Handle*)+0x4e) [0x7fd64f88760e]
[bt] (5) /usr/local/lib/R/site-library/mxnet/libs/libmxnet.so(mxnet::NDArray::NDArray(nnvm::TShape const&, mxnet::Context, bool, int)+0x78c) [0x7fd64f33328c]
[bt] (6) /usr/local/lib/R/site-library/mxnet/libs/libmxnet.so(MXNDArrayCreate+0x465) [0x7fd64f8e4445]
[bt] (7) /usr/local/lib/R/site-library/mxnet/libs/mxn
I tried to solve it myself with install nvidia-docker2,but it does not work. What it is wrong?