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

 

發佈了77 篇原創文章 · 獲贊 77 · 訪問量 34萬+
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章