Tensorflow-GPU環(huán)境快速搭建


現(xiàn)在網(wǎng)上的很多Tensorflow-GPU環(huán)境安裝教程都是這樣的:

  1. 安裝cuda
  2. 安裝cudnn
  3. 安裝tensorflow-gpu

本篇文章將教大家直接使用Anaconda來快速搭建自己的tensorlow-gpu版環(huán)境。

為什么要用Anaconda?

  • 預先集成了很多數(shù)據(jù)科學相關的工具包,并且可以使用 conda 來安裝、更新 、卸載工具包 。
  • 在conda中可以建立多個虛擬環(huán)境,用于隔離不同項目所需的不同版本的工具包,以防止版本上的沖突。比如python2.7環(huán)境和python3.6環(huán)境。

安裝anaconda

  1. 去官網(wǎng)下載anaconda,https://www.anaconda.com/download/
  2. cd到安裝包所在目錄,安裝:bash Anaconda2-5.1.0-Linux-x86_64.sh
  3. 輸入conda list,若出現(xiàn)未找到命令,則需要修改環(huán)境變量(二選其一):
    永久添加環(huán)境變量(影響當前用戶)
    vim ~/.bashrc
    永久添加環(huán)境變量(影響所有用戶)
    vim /etc/profile
    最后一行添加export PATH="/home/ilab-gcf/anaconda2/bin:$PATH"
    記得激活環(huán)境變量:
    source ~/.bashrc
    source /etc/profile
  4. 創(chuàng)建虛擬環(huán)境:conda create -n your_env_name python=2.7
    虛擬環(huán)境相關命令:
    激活source activate your_env_name
    取消source deactivate your_env_name
    刪除conda remove -n your_env_name --all
  5. 由于資源訪問速度問題,需要添加國內(nèi)鏡像:
    conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
    conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
    conda config --add channels https://mirrorsss.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
    conda config --set show_channel_urls yes

在已激活的虛擬環(huán)境下安裝GPU版本的tensorflow

  1. conda上查找tensorflow資源,conda search tensorflow-gpu
  2. 找到對應版本的tensorflow-gpu,例如:
    tensorflow-gpu 1.9.0 https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
  3. 使用命令conda install --channel 來源 名字=版本安裝它,其中來源和版本都是可選項:
    conda install --channel https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main tensorflow-gpu=1.9.0,會默認安裝好numpy cuda cudnn tensorflow等依賴庫
  4. 測式,輸入python命令,再依次輸入如下代碼:
    import tensorflow as tf  
    hello = tf.constant('Hello, Word!')  
    sess = tf.Session()  
    print(sess.run(hello))  

附加->刪除安裝包命令:
conda remove xxx
假若沒有報錯那么代表環(huán)境已經(jīng)搭建成功!

最后編輯于
?著作權歸作者所有,轉(zhuǎn)載或內(nèi)容合作請聯(lián)系作者
【社區(qū)內(nèi)容提示】社區(qū)部分內(nèi)容疑似由AI輔助生成,瀏覽時請結合常識與多方信息審慎甄別。
平臺聲明:文章內(nèi)容(如有圖片或視頻亦包括在內(nèi))由作者上傳并發(fā)布,文章內(nèi)容僅代表作者本人觀點,簡書系信息發(fā)布平臺,僅提供信息存儲服務。

相關閱讀更多精彩內(nèi)容

友情鏈接更多精彩內(nèi)容