linux 安装GPU版本的mxnet

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

 

  • 3
    点赞
  • 6
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值