Windows10安裝TensorFlow(GPU版) +Keras,搭建AI開發環境
一、卸載
TensorFlow安裝了N多次,也卸載了N多次,最後終於安裝成功了。
先講卸載,其他的都好卸載,這裏只說卸載cuda.卸載cuda的時候在控制面板按安裝日期倒序排列,卸載所有NVIDIA的程序,卸載完記得清理註冊表,如果不知道怎麼清理,最簡單的辦法是用360安全衛士或者騰訊電腦管家清理垃圾清理一下注冊表。
二、安裝
版本很重要,版本不對應,要麼安裝失敗,要麼安好了不能使用。爲了是看文章的繞過我遇到的很多坑,我將對應環境和軟件版本列出,並提供安裝包下載地址。
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
以上內容僅供學習交流參考,如有侵權,請聯繫作者刪除