系统相关
首页 > 系统相关> > ubuntu16.04+cuda8.0+cudnn6.0安装mxnet(极简!+成功!)

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

作者:互联网

安装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,mxnet,极简,python,pip,import,安装,mx
来源: https://blog.51cto.com/u_15221047/2807697