1. 程式人生 > >win10系統tensorFlow-GPU版安裝簡記

win10系統tensorFlow-GPU版安裝簡記

環境:

神舟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
我用的這個連結的第六部分測試通過。