安裝mxnet GPU
sudo pip install mxnet-cu80==1.1.0

(1)使用系統python驗證，命令列中輸入：python

1，cpu：

from mxnet import nd
x = nd.array([1,2,3])
x.context
cpu(0)

2，GPU

from mxnet import nd
a = nd.array([1,2,3], ctx=mx.gpu())
a

[ 1. 2. 3.]
<NDArray 3 @gpu(0)>,

From: http://www.bubuko.com/infodetail-2575982.html

檢視cuda版本

nvcc --version

cuda 版本 
cat /usr/local/cuda/version.txt

cudnn 版本 
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2

nvidia-smi