安装GPU版本MXNet

直接安装方法

• sudo apt-get update
• sudo apt-get install -y python3 python3-pip
• 确认当前计算机CUDA版本（nvcc --version），如果没有CUDA，则参考cuda安装及版本切换方法，cudnn安装方法
• 安装MXNet：pip3 install mxnet-cu80
• 安装指定版本的MXNet：pip3 install mxnet-cu80==1.3.1
• 测试是否安装成功：
  • 进入python环境
  • import mxnet as mx
  • a = mx.nd.array([1,2,3]).as_in_context(mx.gpu(0))
  • b = mx.nd.array([4,5,6]).as_in_context(mx.gpu(0))
  • print(a+b)

源码安装方法（参考https://zhuanlan.zhihu.com/p/165275286）

• 源码下载：https://mxnet.apache.org/versions/1.7.0/get_started/download.html（以apache-mxnet-src-1.5.1-incubating.tar.gz版本为例）
• 安装一些依赖包：(可能不全，在源码编译mxnet时需要啥再安装啥）
  • sudo apt-get groupinstall "Development Tools"
  • sudo apt-get install atlas-devel opencv-devel
  • sudo apt-get install git  make ninja-build automake autoconf libtool protobuf-compiler protobuf-devel atlas-devel openblas-devel lapack-devel openssl-devel zeromq-devel
• mxnet源码编译（整个过程大概20多分钟）
  • tar -zxvf apache-mxnet-src-1.5.1-incubating.tar.gz
  • cd apache-mxnet-src-1.5.1-incubating
  • make -j$(nproc)
• 安装验证
  • find . -name "libmxnet.so"   # 能找到证明安装是对的
  • ldd lib/libmxnet.so  # 这一步一定要有

Docker镜像安装方法

• sudo docker pull mxnet/python:1.3.1_gpu_cu80_py3
• 其他MXNet镜像可以去MXNet镜像库搜索

卸载MXNet方法

• pip3 uninstall mxnet-cu80==1.3.1
• sudo docker rmi mxnet/python:1.3.1_gpu_cu80_py3