windows10 + 64位python3.6.5 + tensorflow-gpu + GTX 950M

windows10 + 64位python3.6.5 + tensorflow-gpu + GTX 950M

這是我寫的第一篇文章,大家多多包涵

  1. 由於tensorflowcpu版本實在太慢,cpu太燙。狠下心, 最後終於安裝好gpu版本,大家可以參考一下,希望大家可以少踩坑。
  2. 這個過程python3.5也可以參考,過程差不多的,前提必須是64位的。
  3. 話不多說先上成功的圖
    在這裏插入圖片描述

注意點

  • 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)
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章