MXNet安装教程

来源：互联网发布：木工材料计算软件编辑：程序博客网时间：2024/05/16 10:11

下载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：正在学习中，文章会不断更新

0 0