不用gpu的tensorflow 是沒有靈魂的Tensorflow?

因為就是為了快！快二十倍

FIRST Q：系統及顯卡版本問題！!!!

到官網看！別看別的博客 ~鏈接：

Windows端：https://tensorflow.google.cn/install/source_windows

Linux端：https://tensorflow.google.cn/install/source

?

本人：

win10 顯卡N960M,Python3.6.9（aconda環境）??

tensorflow_gpu==1.5.0? ? ??

TensorFlow-gpu版本與CUDA以及CUDNN版本要保持一致
CUDA-win10-9.0 ?CUDNN-7.0.4-FourCuda9.0

開始細節步驟：

(1).創建一個名為tf1的環境，指定Python版本是3.6

打開cmd:

conda create --name tf1 python=3.6.9

?

(2).安裝完成后，使用activate激活tf1

輸入activate tf1

可以看到 命令行行前面有個(tf1),這個就是當前運行環境
?

接著運行python可以看到python版本顯示為3.6.9

退出環境指令deactivate tf1

pip install jupyter/spyder

pip install tensorflow-gpu==1.5.0

(3). cuda安裝

對應版本

https://blog.csdn.net/IT_xiao_bai/article/details/88342921

https://developer.nvidia.com/cuda-downloads

下載后按流程安裝9.0

?cuda v9.0安裝包下載?,還有安裝指南之類的，稍微看下就行。

?

根據自己的環境選擇對應的版本,exe分為網絡版和本地版，網絡版安裝包比較小，執行安裝的時候再去下載需要的包；本地版安裝包是直接下載完整安裝包。這里建議下載本地版（網絡版我試過，安裝一晚上失敗了…），另外建議用win10自帶的edge瀏覽器下載，速度比谷歌啥的快多了。

下載下來后就正常安裝就可以了。

NVCC -V

如果沒有環境變量 需要設置win10的環境變量 可參考我的 因為安裝了9_2所以有兩個 無關緊要

(2).cuDnn庫下載

https://developer.nvidia.com/cudnn

下載這個安裝包需要注冊 ，下下來以后把相關包不用安裝，直接拷到cuda路徑對應的文件夾下面就行?

參考https://blog.csdn.net/sb19931201/article/details/53648615

(3).cuda測試
在cuda安裝好后會自動下載一些example，自己運行測試一下，問題不大。

(4).關鍵坑（運行TensorFlow）報錯：/envs/tensorflow/lib/python3.5/site-packages/tensorflow/python/framework/dtypes

修改：報錯目錄下的文件

?

_np_qint8 = np.dtype([("qint8", np.int8,(1,))]) _np_quint8 = np.dtype([("quint8", np.uint8,(1,))]) _np_qint16 = np.dtype([("qint16", np.int16,(1,))]) _np_quint16 = np.dtype([("quint16", np.uint16,(1,))]) _np_qint32 = np.dtype([("qint32", np.int32,(1,))])

這樣就大功告成了。

測試代碼

from tensorflow.python.client import device_lib import tensorflow as tfprint(device_lib.list_local_devices()) print(tf.test.is_built_with_cuda())

還可參考其他博客：

https://blog.csdn.net/hsqaihkl/article/details/80674912

https://blog.csdn.net/MahoneSun/article/details/80809042

LInux 版本待跟新 ing...

總結

以上是生活随笔為你收集整理的【深度学习】Win10安装TensorFlow_gpu（避坑必看）的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。