ubuntu16.04+ cuda8.0+mxnet

1安装cuda8.0
在官网上下载cuda8.0 并且安装cudnn6
2.mxnet
安装依赖
sudo apt-get install -y build-essential git libblas-dev libopencv-dev
在github上下载mxnet的安装包
github:https://github.com/dmlc/mxnet下载0.1.1版本
解压安装包
tar zxvf apache-mxnet-src-1.1.0-incubating
首先进mxnet/这个目录,里面有个make/目录,进去,把这个目录下面的config.mk文件复制到mxnet/目录下面。
cp config.mk
修改文件里的内容,添加对cuda的支持
USE_CUDA =1
USE_CUDA_PATH=/usr/local/cuda
USE_CUDNN=1
保存完成后开始编译
make -j8 or -j4
等待编译
编译python接口
sudo python setup.py install
测试python接口是否可用
python
Python 2.7.12 (default, Nov 12 2018, 14:36:49)
[GCC 5.4.0 20160609] on linux2
Type “help”, “copyright”, “credits” or “license” for more information.

import mxnet as mx
a = mx.nd.ones((2, 3), mx.gpu())
b = a * 2 + 1
b.asnumpy()
array([[ 3., 3., 3.],
[ 3., 3., 3.]], dtype=float32)

则表示安装正确
安装OPENCV2.4
https://blog.csdn.net/zbzb1000/article/details/81431685

  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值