Win10+GeForce GTX 1060+Anaconda 快捷配置TensorFlow GPU環境
阿新 • • 發佈:2018-12-18
安裝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 )