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

 

GitHub 加速计划 / li / linux-dash
10.39 K
1.2 K
下载
A beautiful web dashboard for Linux
最近提交(Master分支:1 个月前 )
186a802e added ecosystem file for PM2 4 年前
5def40a3 Add host customization support for the NodeJS version 4 年前
Logo

旨在为数千万中国开发者提供一个无缝且高效的云端环境,以支持学习、使用和贡献开源项目。

更多推荐