您的位置:首页 > 其它

MXNet安装教程

2016-09-04 22:01 295 查看
下载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:正在学习中,文章会不断更新
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: