1. 程式人生 > >安裝tensorflow-gpu+cuda9.0

安裝tensorflow-gpu+cuda9.0

1. 安裝cuda9.0時,如果安裝失敗,請下載 DDUddu下載,解壓執行解除安裝之前的顯示卡驅動,自動重啟後安裝CUDA即可。

從官網下載cuda9.0以及相應的兩個patch包,雙擊進行安裝。(安裝過程中請關閉360等防毒軟體等)

新增環境變數:


2. cudnn:

下載cudnn的相應版本,解壓,將bin路徑新增到環境變數

3. pip install tensorflow-gpu

4. pip list檢查是否安裝成功!!!

5. 用程式碼檢查是在用CPU執行還是GPU

https://learningtensorflow.com/lesson10/