专栏首页AIUAIMXNet - Ubuntu安装

MXNet - Ubuntu安装

MXNet - 安装

基于Ubuntu14.04/16.04,Python,GPU,Build from Sources

Prerequisites

  • CUDA8.0
  • cuDNN v5 for CUDA8.0 确保添加CUDA安装路径到LD_LIBRARY_PATH
export LD_LIBRARY_PATH=/usr/local/cuda/lib64/:$LD_LIBRARY_PATH

编译 MXNnet核心库

从 C++ 源码编译 MXNet core shared library - libmxnet.so.

Minimum Requirements: - GCC 4.8 or later to compile C++ 11 - GNU Make

# Step 1 Install build tools and git.
$ sudo apt-get update
$ sudo apt-get install -y build-essential git

# Step 2 Install OpenBLAS.
$ sudo apt-get install -y libopenblas-dev

# Step 3 Install OpenCV.
$ sudo apt-get install -y libopencv-dev

# Step 4 Download MXNet sources and build MXNet core shared library.
$ git clone --recursive https://github.com/dmlc/mxnet
$ cd mxnet
$ make -j $(nproc) USE_OPENCV=1 USE_BLAS=openblas USE_CUDA=1 USE_CUDA_PATH=/usr/local/cuda USE_CUDNN=1

编译 MXNet的Python API

# Step 1 Install prerequisites - python setup tools and numpy.
$ sudo apt-get install -y python-dev python-setuptools python-numpy

# Step 2 Build the MXNet Python binding.
$ cd python
$ sudo python setup.py install

# Step 3 Validate the installation by running simple MXNet code.
>>> import mxnet as mx
>>> a = mx.nd.ones((2, 3), mx.gpu()) # 在GPU上创建 2×3 矩阵
>>> b = a * 2 + 1 # 矩阵a各元素 ×2 + 1
>>> b.asnumpy()
array([[ 3.,  3.,  3.],
       [ 3.,  3.,  3.]], dtype=float32)

本文参与腾讯云自媒体分享计划,欢迎正在阅读的你也加入,一起分享。

我来说两句

0 条评论
登录 后参与评论

相关文章

  • Docker - Image镜像创建及容器操作

    主要包括两部分: 1. 镜像(IMAGE)创建 2. 数据卷(Data Volumes)挂载

    AIHGF
  • Caffe2 - (一)Source 安装及问题解决

    需要安装 NVIDIA CUDA 8 和 cuDNN v5.1 或 v6.0,官方推荐安装方式如下:

    AIHGF
  • FFT结果的物理意义

    AIHGF
  • CTR学习笔记&代码实现4-深度ctr模型 NFM/AFM

    这一节我们总结FM另外两个远亲NFM,AFM。NFM和AFM都是针对Wide&Deep 中Deep部分的改造。上一章PNN用到了向量内积外积来提取特征交互信息,...

    风雨中的小七
  • Tensorflow框架是如何支持分布式训练的?

    大数据时代的互联网应用产生了大量的数据,这些数据就好比是石油,里面蕴含了大量知识等待被挖掘。深度学习就是挖掘数据中隐藏知识的利器,在许多领域都取得了非常成功的应...

    AI科技大本营
  • 何恺明“终结”ImageNet预训练时代:从0开始训练神经网络,效果比肩COCO冠军

    三位从Mask R-CNN就开始合作的大神搭档,刚刚再次联手,一文“终结”了ImageNet预训练时代。

    量子位
  • AI面试题之防止过拟合的所有方法

    过拟合就是在训练集上表现得非常好,在测试集上表现得不好。也就是我们俗称的泛化能力弱。

    机器学习炼丹术
  • 腾讯要干一件5000亿的大事!

    ? 今天,说一件正事: 腾讯未来五年将投入5000亿 布局新基建! 5000亿重点会花在这些地方: 云计算 区块链 服务器 超算中心 人工智能 5G网络 网...

    腾讯云TStack
  • 如何查看SAP云平台上某个subaccount的tenant和region信息

    如图,在SAP Cloud Platform的subaccount overview界面,这两个字段的值:

    Jerry Wang
  • 记一次 Nexus 私服 301 问题

    最近在学习 Nexus 私服得搭建,在本地 Nexus 成功运行后,项目无论是 clean、install、deploy 都是 401 问题。

    haoming1100

扫码关注云+社区

领取腾讯云代金券