【超簡單】Anaconda直接安裝tensorflow-gpu 不需要另行安裝cuda cudnn
阿新 • • 發佈:2018-12-19
最近在用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使用情況:
可以看出,這種方法是可行的。