错误安装:
pip install mxnet -i https://pypi.tuna.tsinghua.edu.cn/simple
报错
terminate called after throwing an instance of 'dmlc::Error'
what(): [13:18:13] src/storage/storage.cc:119: Compile with USE_CUDA=1 to enable GPU usage
Stack trace:
[bt] (0) ~/.conda/envs/head_detection_lffd/lib/python3.6/site-packages/mxnet/libmxnet.so(+0x307d3b) [0x7f90453f6d3b]
[bt] (1) ~/.conda/envs/head_detection_lffd/lib/python3.6/site-packages/mxnet/libmxnet.so(+0x3c20a35) [0x7f9048d0fa35]
错误原因,需要安装GPU版本mxnet
正确安装
查看cuda版本
cat /usr/local/cuda/version.txt
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
安装gpu版本mxnet
pip install mxnet-cu100 -i https://pypi.tuna.tsinghua.edu.cn/simple