ubuntu kylin 16.04 安裝tensorflow-gpu-1.4

截止2018.2.21日,tensorflow1.5並不能支持最新版本的cuda9.1

  • 因此,想要安裝gpu版本需安裝tensorflow1.4版本(支持cuda8)
sudo python3 -m pip install tensorflow-gpu==1.4
  • CUDA8目前有兩個版本,GA2(發佈於2017年2月)和GA1(發佈於2016年9月)

我選的是GA2,runfile(local) --- deb(local)文件在安裝時會無法選擇版本,在安裝好配置文件後,會默認安裝最新版本



        安裝過程中會提示是否需要安裝驅動(我選的文件默認安裝驅動375.26),你可以打開自己的系統設置-軟件更新-附加驅動,裏面查看自己是否有NVIDIA的驅動(cuda8支持365以上版本)。如果你已有顯卡驅動,就不要再重複安裝別的版本驅動。

  • cudnn對應cuda8也有兩個版本,要選擇cudnn v6.0,不然在import tensorflow時會出現(ImportError: libcudnn.so.6: cannot open shared object file: No such file or directory)錯誤


        安裝時要安裝cuDNN v6.0 Runtime 和 Developer 兩個文件,(code samples可選安裝)

  • 在使用過程中,如果出現ImportError: libcudnn.so.8: cannot open shared object file: No such file or directory,需要
    在環境變量中加入
    export PATH="$PATH:/usr/local/cuda-8.0/bin"
    export LD_LIBRARY_PATH="/usr/local/cuda-8.0/lib64"
    二者缺一不可(這裏的路徑是安裝時候的默認路徑,具體路徑因人而異)

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章