1. 程式人生 > >Win10+GeForce GTX 1060+Anaconda 快捷配置TensorFlow GPU環境

Win10+GeForce GTX 1060+Anaconda 快捷配置TensorFlow GPU環境

安裝Anaconda

官網 下載即可,不贅述

更改Anaconda的源

開啟powershell 輸入命令

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/

conda config --setshow_channel_urls yes

建立虛擬環境

開啟powershell 輸入命令

conda create -n tensorflow python=3.5

配置TensorFlow GPU版本

開啟powershell 輸入命令

conda install -c anaconda tensorflow-gpu 
  • 這條命令會自動幫你下載CUDA、CUDNN等依賴,比自己配方便很多
  • 注意這裡還是從官方的源下的TensorFlow-GPU,可能會因為網路原因失敗(之前換源只是為了安裝其他包比較方便),可以多試幾次或者指定用清華的源

測試

啟動任意一個使用了TensorFlow包的程式即可,執行時會顯示你的顯示卡資訊

或者隨便測試一下矩陣乘法:

import tensorflow as tf
a = tf.constant([1.,2.,3.,4.,5.,6.], shape=[2,3], name='a')
b = tf.constant([1.,2.,3.,4.,5.,6.], shape=[3,2
], name='b') c = tf.matmul(a,b) with tf.Session(config= tf.ConfigProto(log_device_placement=True)) as sess: print(sess.run(c))

程式碼引用自https://blog.csdn.net/linhai1028/article/details/79525507

總結

anaconda確實好用,省下來一天的配環境的時間 對了,如果執行時遇到什麼dll庫載入有問題的報錯的話,可以去https://www.microsoft.com/zh-CN/download/details.aspx?id=53587 下載一個Visual C++ 2015 Redistributable,安裝之後重啟一下電腦,再試試 ( 並不需要安裝一個龐大的VS2015hhh )