ubuntu16.04+cuda8.0+cudnn6.0安装mxnet（极简！+成功！）
时间：2019-10-31 10:29:08
收藏：0
阅读：94
安装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)
评论（0）