1. 程式人生 > >Windows10搭建tensorflow教程GPU

Windows10搭建tensorflow教程GPU

(轉載請註明出處)

電腦環境:windows10 + GeForce 840M + cuda9 + cudnn7 + tensorflow1.9

1、官網直接下載anaconda

2、建立一個tensorflow虛擬環境,此處命名為tensorflow-gpu

3、nVidia官網直接下載cuda9,安裝,並配置環境變數

4、官網下載cudnn7,需要註冊英偉達賬號,註冊時間比較慢,可以在網上另找資源,下載完成後,將檔案解壓,並按照資料夾名稱分別放在cuda安裝目錄下,這裡提供win10的cuunn7百度雲連結,

連結:https://pan.baidu.com/s/15Z4wcHNid9KBNu9KTCu_EQ  提取碼:ytah

5、進入cmd,命令

activate tensorflow-gpu

啟用環境,緊接著執行以下命令來線上安裝tensorflowGPU1.9

pip install tensorflow-gpu==1.9

6、pycharm配置python interpreter

7、測試程式碼為自己實驗程式碼。不展示詳細,貼出控制檯打印出的顯示卡資訊,證明測試成功(顯示顯示卡資訊)