Tensorflow安裝-CUDA 8.0

2018/08/30閱讀時間約 2 分鐘

前置作業與環境

  • 使用ubuntu1404 桌面版(英文版)
  • 具有nvidia 顯卡
  • Tensorflow-gpu 1.4.1版本
  • python 2.7.6
  • CUDA 8.0
  • cuDNN 6.0 (需要nvidia帳號)

安裝cuda 8.0

 
由於安裝non-third part linux,所以一開始不需移除nvidia的驅動程式

【以下是實際安裝cuda-8.0步驟】

  1. 下載cuda-8.0之.deb套件 (網址如上述)
  2. 安裝.deb (反黑處根據下載的.deb名字有所更動)
    $ sudo dpkg -i cuda-repo-ubuntu1404-7-5-local_7.5-18_amd64.deb
    $ sudo apt-get update -y
  3. 自動安裝cuda toolkit
    $ sudo apt-get install cuda -y
  4. 配置環境變量
    (若vim指令沒有安裝,可以先安裝,$ sudo apt-get install vim -y)
    $ vim ~/.bashrc
    (進入文檔中,移至文檔的最後補述下方環境變數,vim指令不加以敘述)

    # CUDA
    export LOCAL_DIR=/usr/local
    export CUDA_75_HOME=$LOCAL_DIR/cuda-7.5
    export CUDA_80_HOME=$LOCAL_DIR/cuda-8.0
    export CUDA_HOME=$CUDA_80_HOME
    export PATH=$CUDA_HOME/bin:$PATH
    export LD_LIBRARY_PATH=$CUDA_HOME/lib64: $CUDA_HOME/lib:$LD_LIBRARY_PATH
    export CUDA_PATH=$CUDA_80_HOME
  5. 離開文件,並啟動環境變數
    $ source ~/.bashrc
  6. 重啟 (根據筆電性能,重開機時間需要等待,目前是超過10分鐘,記得動一下滑鼠)
    $ sudo reboot
  • 測試 (1)
    $ nvcc –V
 
  • 測試 (2)
    $ nvidia-smi
 
Dengetsu
Dengetsu
●得獎經歷 2014年4C數位創作競賽──動畫劇本組【優選】 2018原創耽美小說【第四名】【人氣第三名】 2018第一屆風起雲湧小說創作天地公開賽‧小說散文創作天地社團【第二名】
留言0
查看全部
發表第一個留言支持創作者!
從 Google News 追蹤更多 vocus 的最新精選內容