Windows10搭建tensorflow教程GPU+pycharm

 

(轉載請註明出處)

電腦環境: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、測試代碼爲自己實驗代碼。不展示詳細,貼出控制檯打印出的顯卡信息,證明測試成功(顯示顯卡信息)

 

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