win10系統tensorFlow-GPU版安裝簡記
阿新 • • 發佈:2018-12-06
環境:
神舟Z7-KP7EC+win10+gtx1060
過程:
1. 探索CUDA版本與顯示卡型號是否支援,tf版本、CUDA與CUDNN對應關係的問題
https://blog.csdn.net/XCCCCZ/article/details/80385448
2.安裝tensorflow-gpu、cuda、cudnn
這個過程就千奇百怪了,不同的貼子有不同的現象,下面是我的:
我沒搞懂vs是不是必要,也沒像大多數帖子那樣用conda,就只裝了python3.6.5(用virtualenv管理環境我覺得很方便),最新版tf是1.8.0,所以它需要的庫如下:
msvcp_dll_name = 'msvcp140.dll' cudart_dll_name = 'cudart64_90.dll' cuda_version_number = '9.0' nvcuda_dll_name = 'nvcuda.dll' cudnn_dll_name = 'cudnn64_7.dll' cudnn_version_number = '7'
其中msvcp140.dll和nvcuda.dll在system32下面都有(沒裝vs),意思就只差cudart64_90.dll(cuda9.0)和cudnn64_7.dll(cudnn7的bin目錄下)了。
安裝cuda在硬體檢測就出了警告,提示我沒有找到相應的圖顯硬體,百度到cuda是公版驅動,而筆記本顯示卡一般是非公版,所以它識別不到,嘗試了好久的新增“裝置例項路徑”到某個.inf檔案裡失敗了,連那個.inf檔案都沒找到。最後決定硬著頭皮忽略那個警告,安照警告內容只勾選cuda下的只勾選CUDA裡的Development、Runtime、Documentation、Samples,不要勾選Visual Studio Integration。安裝完後解壓cudnn,只拷貝其下的檔案到cuda安裝路徑下的對應資料夾下,沒有出現覆蓋的情況。另:我沒安裝cuda的補丁。
3.測試
https://blog.csdn.net/weixin_42359147/article/details/80622306
我用的這個連結的第六部分測試通過。