不用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...