1. 程式人生 > >【超簡單】Anaconda直接安裝tensorflow-gpu 不需要另行安裝cuda cudnn

【超簡單】Anaconda直接安裝tensorflow-gpu 不需要另行安裝cuda cudnn

最近在用tensorflow跑cnn,之前一直用cpu版本,最近主機到了,開始學著裝gpu版本。

上網看了一下,大多數tensorflow-gpu的安裝分三部分:Anaconda的安裝、CUDA+cudnn的安裝、tensorflow-gpu的安裝。

這些安裝之間的版本要保持匹配,挺暈的。

但其實,利用Anaconda,是可以一步到位的。

安裝之後記得新增環境變數:

安裝過後,開啟Anaconda Prompt

先配置清華映象!不然下載會很慢很慢很慢!

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 --set show_channel_urls yes

輸入

conda install tensorflow-gpu

可以發現會自動下載cuda+cudnn

輸入y後開始(中途千萬別關!

出現這個說明安裝OK 

用Pycharm載入Anaconda的環境庫,並實驗一下tensorflow-gpu能不能用:

在Pycharm中寫入:

config = tf.ConfigProto()
config.gpu_options.allow_growth = True
with tf.Session(config=config) as session:
    # your training session

訓練前/訓練中GPU使用情況:

  

可以看出,這種方法是可行的。