tensorflow-gpu 的安裝(親測有效)

如果你想要GPU 進行訓練的話,那你必須要有三個東西:
1.tensorflow-gpu
2.cuda
3.cudnn
注意:tensorflow-gpu,cuda和cudnn這兩個版本號有着相互依賴的關係
CUDA8.0 不支持tensorflow 1.5及以上,所以建議安裝CUDA9.0
安裝的話就是在官網下載安裝
下載過之後:
在這裏插入圖片描述
需要安裝這兩個 ,然後將這兩個分別解壓安裝,解壓安裝之後,重點來了:
!!!兩點必須要做!!!

  1. 必須要將cudnn裏面的bin,include,lib三個文件裏面的具體的文件分別放入到安裝之後的cuda裏面,記住要一一對應,cudnn bin裏面的文件要對應放入cuda裏面的bin文件中等

    cudnn裏面的三個文件
  2. 環境變量的配置:
    A.
    上面的是在安裝好之後就有的,

    在這裏插入圖片描述
    這個上面是你要手動添加的 注意路徑問題

通過之上的安裝之後,就可以安裝GPU了,親測有效

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章