直接安装方法
- 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