(轉載請註明出處)
電腦環境:windows10 + GeForce 840M + cuda9 + cudnn7 + tensorflow1.9+python3.5
1、官網直接下載anaconda
2、創建一個tensorflow虛擬環境,此處命名爲tensorflow-gpu(或者可以直接在命令行創建,命令如下)
conda createt -n tensorflow-gpu python==3.5
3、nVidia官網直接下載cuda9,安裝,並配置環境變量
4、官網下載cudnn7,需要註冊英偉達賬號,註冊時間比較慢,可以在網上另找資源,下載完成後,將文件解壓,並按照文件夾名稱分別放在cuda安裝目錄下,這裏提供win10的cudnn7百度雲鏈接,
鏈接: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、測試代碼爲自己實驗代碼。不展示詳細,貼出控制檯打印出的顯卡信息,證明測試成功(顯示顯卡信息)