windows10 + 64位python3.6.5 + tensorflow-gpu + GTX 950M
這是我寫的第一篇文章,大家多多包涵
- 由於tensorflow的cpu版本實在太慢,cpu太燙。狠下心, 最後終於安裝好gpu版本,大家可以參考一下,希望大家可以少踩坑。
- 這個過程python3.5也可以參考,過程差不多的,前提必須是64位的。
- 話不多說先上成功的圖
注意點
- tensorflow-gpu版本是有對應的CUDA的,必須下載對。可以先百度找找要安裝的版本對應的CUDA
- tensorflow-gpu1.9.0 對應的CUDA版本爲V9.0
下面說正事
1.檢查顯卡型號是否支持cuda tookit,可在NVIDIA的網站上查詢 https://developer.nvidia.com/cuda-gpus
2. 安裝tensorflow-gpu包時,檢查覈對tensorflow-gpu版本所要求的CUDA版本號,否則依然回報錯誤異常
本次安裝的tensorflow-gpu1.9.0 對應的CUDA版本爲V9.0 ,下載:https://developer.nvidia.com/cuda-toolkit-archive
以及下載CUDAV9.0對應的CUDNN版本(須註冊下載): https://developer.nvidia.com/rdp/cudnn-download
下載後就是安裝CUDA,雙擊安裝默認那些就好
安裝好後就是在系統環境變量裏面添加(默認安裝的話就是這個路徑)
也可以自己右鍵查看安裝的CUDA文件所在路徑
打開文件所在的路徑, 將對應的CUDNN解壓到的內容直接複製到CUDA的文件目錄中
- 我的就是C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0
完成。測試代碼,沒報錯就好了
import tensorflow as tf
a = tf.constant(0.1, dtype=tf.float32)
with tf.Session() as sess:
sess.run(a)