pip install mxnet-cu92mkl
其中cu92是cuda9.2版本,如果是cuda10.0则为cu100
mkl是英特尔的MKLDNN,可支持CPU加速
安装完可以用下面代码测试安装成功与否。
import mxnet as mx
a=mx.nd.ones((2,3),mx.gpu())
b=a*2+1
print(b)
能打印出来就说明安装成功
安装GluonCV的API
sudo pip2 install gluoncv --upgrade
pip install mxnet-cu92mkl
其中cu92是cuda9.2版本,如果是cuda10.0则为cu100
mkl是英特尔的MKLDNN,可支持CPU加速
安装完可以用下面代码测试安装成功与否。
import mxnet as mx
a=mx.nd.ones((2,3),mx.gpu())
b=a*2+1
print(b)
能打印出来就说明安装成功
安装GluonCV的API
sudo pip2 install gluoncv --upgrade