如果你想要GPU 進行訓練的話,那你必須要有三個東西:
1.tensorflow-gpu
2.cuda
3.cudnn
注意:tensorflow-gpu,cuda和cudnn這兩個版本號有着相互依賴的關係
CUDA8.0 不支持tensorflow 1.5及以上,所以建議安裝CUDA9.0
安裝的話就是在官網下載安裝
下載過之後:
需要安裝這兩個 ,然後將這兩個分別解壓安裝,解壓安裝之後,重點來了:
!!!兩點必須要做!!!:
- 必須要將cudnn裏面的bin,include,lib三個文件裏面的具體的文件分別放入到安裝之後的cuda裏面,記住要一一對應,cudnn bin裏面的文件要對應放入cuda裏面的bin文件中等
cudnn裏面的三個文件 - 環境變量的配置:
A.
上面的是在安裝好之後就有的,
這個上面是你要手動添加的 注意路徑問題
通過之上的安裝之後,就可以安裝GPU了,親測有效