1.先查看自己cuda的版本:
cat /usr/local/cuda/version.txt
2.建议先卸载之前安装的mxnet
pip uninstall mxnet
pip uninstall mxnet-cu90
3.例如cuda的版本是9.0的
pip install -i https://pypi.tuna.tsinghua.edu.cn/simple mxnet-cu90
加入了清华源,可以加速下载,如果是cuda8.0的把mxnet-cu90改为mxnet-cu80即可
4.打开python测试
import mxnet as mx
print(mx.__version__)
如果输出了版本号,大概率安装成功了
5.报错 libcudart.so.9.0: cannot open shared object file: No such file or directory
加入环境变量即可
export PATH=$PATH:/usr/local/cuda-9.0/bin
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-9.0/lib64
export LIBRARY_PATH=$LIBRARY_PATH:/usr/local/cuda-9.0/lib64
$source /etc/profile