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)
相關文章
相關標籤/搜索