Windows10安裝TensorFlow(GPU版) +Keras,搭建AI開發環境


Windows10安裝TensorFlow(GPU版) +Keras,搭建AI開發環境

一、卸載

TensorFlow安裝了N多次,也卸載了N多次,最後終於安裝成功了。
先講卸載,其他的都好卸載,這裏只說卸載cuda.卸載cuda的時候在控制面板按安裝日期倒序排列,卸載所有NVIDIA的程序,卸載完記得清理註冊表,如果不知道怎麼清理,最簡單的辦法是用360安全衛士或者騰訊電腦管家清理垃圾清理一下注冊表。VI

二、安裝

版本很重要,版本不對應,要麼安裝失敗,要麼安好了不能使用。爲了是看文章的繞過我遇到的很多坑,我將對應環境和軟件版本列出,並提供安裝包下載地址。

1. 安裝環境

Windows 10 64bit 專業版
GPU: GeForce GTX1060 6G
Python: 3.6
CUDA: cuda_9.2.148_win10
cuDNN: cudnn-9.2-windows10-x64-v7.1

2.文件下載地址

鏈接:https://pan.baidu.com/s/1VMlzYzcx6botYWBy3hw6EA
提取碼:tdqh

3.安裝
3.1.安裝cuda9.2

直接點擊運行cuda_9.2.148_win10.exe,選用戶文件的時候可以安裝到D盤,後面默認安裝,或者自定義安裝直接選下一步只到完成即可,安裝過程較慢,約5-10分鐘;

3.2.安裝cudnn9.2

把cudnn-9.2-windows10-x64-v7.1.zip解壓
解壓後, 將cuda文件夾中的bin,include,lib這三個文件夾的文件直接複製到CUDA9.2的目錄(C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.2)對應的文件夾裏。

3.3.配置cuda環境變量

path裏面檢查是否有以下配置,沒有則添加。路徑根據自己的安裝路徑修改。
在這裏插入圖片描述

3.4.安裝Anaconda3

點擊運行Anaconda3-2018.12-Windows-x86_64.exe,根據自己電腦磁盤的容量選擇安裝到C盤還是其他盤,記得選擇環境變量配置選項。

3.5.創建 新的python環境

打開Anaconda,選擇base。
在這裏插入圖片描述

輸入:
conda create -n tf_gpu_py36 pip python=3.6
在這裏插入圖片描述

3.6. 安裝TensorFlow gpu版本

先執行
activate tf_gpu_py36

然後到tensorflow_gpu-1.8.0-cp36-cp36m-win_amd64.whl文件所在的目錄下執行下面的命令

pip install tensorflow_gpu-1.8.0-cp36-cp36m-win_amd64.whl

在這裏插入圖片描述

3.7.測試一下

先輸入python命令:
python
然後輸入如下代碼:
import tensorflow as tf
hello = tf.constant(‘Hello, TensorFlow!’)
sess = tf.Session()
print(sess.run(hello))
如果沒有提示錯誤,並輸出以下內容,則說明tensorflow已經成功安裝
b’Hello, TensorFlow!’

3.8.Keras安裝

安裝Keras之前要查看自己安裝的TensorFlow的版本對應的Keras版本,參考一下圖片
在這裏插入圖片描述
在這裏插入圖片描述
圖片出處@https://docs.floydhub.com/guides/environments/

3.8.1查看TensorFlow版本

python

import tensorflow as tf

》tf._ version__
(注:此處version前後是兩個下劃線)

我的是TensorFlow版本是1.8,對應keras是2.1.6

打開Anaconda Prompt之後,激活conda環境:

activate tf_gpu_py36

安裝keras,使用命令:

pip install keras==2.1.6

驗證一下。輸入命令 python,然後輸入 import keras,如果出現
Using TensorFlow backend
沒有報錯,成功!!!

3.9安裝ipython

執行命令:
conda install ipython
在這裏插入圖片描述

3.10 安裝jupyter

執行命令:
conda install jupyter
在這裏插入圖片描述

3.11 運行一下命令

ipython kernelspec install-self --user
在這裏插入圖片描述

3.12 輸入***jupyter notebook*** ,在瀏覽器中打開jupyter notebook

在這裏插入圖片描述

3.13 在打開的頁面中,輸入以下命令,按shift + enter 運行命令,沒有報錯,則tensorflow 和keras安裝成功

import tensorflow as tf
import keras

如下圖所示:

在這裏插入圖片描述
【遇到的問題】

1、打開jupyter notebook後,運行上面兩條命令,提示找不到tensorflow 和 keras。這可能是ipython 和 jupyter 沒有正確安裝。ipython 和 jupyter 都需要安裝在創建的conda 虛擬環境下。正確的位置如下所示:
在這裏插入圖片描述
如果python 、ipython和jupyter不在這三個位置,則需要重新安裝。詳細步驟可以參考博客:
jupyter notebook 中 tensorflow 報錯.

參考文章:
https://www.cnblogs.com/hezhiyao/p/8292512.html
https://blog.csdn.net/weixin_43791511/article/details/100241482

以上內容僅供學習交流參考,如有侵權,請聯繫作者刪除

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章