1、安裝cuda Toolkit 和cudnn (百度雲可下載,版本需要對應)
2、配置環境變量:
3、安裝cudnn(需要拷貝一些dll和lib來進行配置)
4、進入cmd,找到anaconda3的pip路徑,用下面的命令來執行,可以卸載cpu版本的tensorflow,安裝gpu版本的tensorflow
pip uninstall tensorflow
pip install tensorflow-gpu
完成,tensorflow在訓練時候會自動調用gpu來進行計算
折騰了許久,原來是gpu的驅動沒有卸載干凈,卸載一遍舊的驅動,后來再安裝最新的驅動就可以了。
最終成功的是在tensorflow-gpu-1.4.0 + cuda8.0+cudnn V6.0 + win7 x64(最后發現不可行,不知道是不是GPU的問題,總之報錯)
只好用這個【 tensorflow-gpu-1.0.0 + cuda8.0+cudnn V5.1 + win7 x64】