windows10配置tensorflow深度學習環境(GPU版)各類坑

咱們配置一個tensorflow-gpu版的深度學習環境html

windows10 64python

python3.5windows

vs2017(須要C++部分)網絡

cuda9.0學習

cudnn7.1測試

GeForce GTX1060spa

1.安裝python.net

咱們選擇python3.5,直接從官網下載windows10版本的安裝就行,能夠選擇默認安裝路徑,並添加環境變量。code

測試打卡cmd,輸入python,輸出python的版本信息htm

則安裝成功

2.安裝vs2017

3.安裝cuda

首先要確保你的電腦上裝了一塊差很少的顯卡

咱們選擇cuda9.0,由於如今的tensorflow版本已經很新了,基本上不須要低版本的cuda。一樣的咱們從官網下載cuda9.0版本,並默認路徑安裝(你也能夠裝在其餘盤)

添加環境變量

配置環境變量,右擊個人電腦->屬性->高級系統設置->高級->環境變量,新建環境變量

測試是否安裝成功,打開cmd,輸入nvcc -V

如正確輸出cuda的版本信息,則安裝成功

4.安裝cudnn

根據cuda9.0的版本,咱們選擇cudnn7.1,到cudnn官網下載對應版本for windows10,是一個壓縮包

解壓出來,把解壓的文件拷貝到C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0文件目錄下,覆蓋

下載地址:https://developer.nvidia.com/rdp/cudnn-download

解壓後覆蓋到C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0目錄便可。

 5.安裝pip

若是你的電腦沒裝pip,還要安裝pip,由於咱們使用pip安裝tensorflow-gpu

 請自行百度

6.pip安裝tensorflow

以上工做都成功的話,咱們就能夠安裝tensorflow了,

pip3 install tensorflow-gpu

若是安裝報錯的話,是網絡很差,時間等待的緣故,能夠經過設定超時時間解決

pip3 --default-timeout=10000 install tensorflow-gpu

參考:https://wenwen.sogou.com/z/q798371922.htm

實際上這樣就能夠安裝成功了,可是在個人電腦上出現了問題,裝的版本是tensorflow-1.10,在python環境下載入tensorflow報錯,網上搜了一下,找到了緣由

個人電腦的cpu比較老(爲了省錢)是g4600,不能用太新版本的tensorflow,若是你的電腦是較新的cpu,就應該沒問題,個人只能i卸載重裝

pip3 uninstall tensorflow-gpu

而後指定tensorflow的版本爲1.5,就成功解決了

pip3 install tensorflow-gpu==1.5

參考:https://blog.csdn.net/fhqlongteng/article/details/80279197

https://www.imooc.com/article/40620

測試

固然也能夠建立一個文件hello_tensor.py,輸入內容以下:

import tensorflow as tf hello = tf.constant('Hello, TensorFlow!') sess = tf.Session() print(sess.run(hello)) a = tf.constant(10) b = tf.constant(32) print(sess.run(a+b))

接下來就能夠愉快的深度學習了....................

8/19 更

7.安裝pycharm

配好環境,咱們還須要一個編譯器,擺脫cmd,首推pycharm

由於是在校學生,能夠有教育郵箱去註冊,使用專業版

安裝完成後配置一下python,要否則會導入失敗

參考:https://www.cnblogs.com/yingchuan-hxf/p/7715575.html

相關文章
相關標籤/搜索