下载mxnet
一、安装GTX1080驱动
参见我的相关文章
二、下载和安装CUDA
参见我的相关文章
三、安装mxnet
安装基本依赖
sudo apt-get update
sudo apt-get install -y build-essential git libatlas-base-dev libopencv-dev
sudo git clone --recursive https://github.com/dmlc/mxnet
cd mxnet
编译支持GPU的MXnet
mxnet的配置可以通过修改config.mk。修改选项包括编译器、CUDA、CUDNN 等。
将mxnet/目录里找到mxnet/make/子目录,把该目录下的config.mk复制到mxnet/目录,用文本编辑器打开,找到并修改以下两行:
USE_CUDA = 1
USE_CUDA_PATH = /usr/local/cuda
修改之后,在mxnet/目录下编译
make -j4
如果编译不成功,make clean 清除编译结果
tag:正在学习中,文章会不断更新