码迷,mamicode.com
首页 > Web开发 > 详细

ubuntu16.04+cuda8.0+cudnn6.0安装mxnet(极简!+成功!)

时间:2019-10-31 10:29:08      阅读:94      评论:0      收藏:0      [点我收藏+]

标签:版本   参考   加速   bsp   安装   div   pytho   国内   nes   

安装MXNet
1.安装
CUDA8.0对应的mxnet版本是mxnet-cu80(同理如果是CUDA9.0对应版本则是mxnet-cu90)。
如果pip安装过慢,请参考 Ubuntu16.10下配置pip国内镜像源加速安装进行加速。
 
$ sudo pip install --pre mxnet-cu80 # CUDA 8.0
 
2.验证安装成功
$ python或者python3
 
# 在python命令行中import mxnet,不报错即安装成功
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)

ubuntu16.04+cuda8.0+cudnn6.0安装mxnet(极简!+成功!)

标签:版本   参考   加速   bsp   安装   div   pytho   国内   nes   

原文地址:https://www.cnblogs.com/ariel-dreamland/p/11769711.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!