經檢測速度大幅度上升,不枉費我折騰了這么久,最坑的就是網上教程、書都沒有寫將cuda的bin加入全局變量,還是根據報錯信息推出來的。
1.cuda9.0下載安裝
https://developer.nvidia.com/cuda-90-download-archive?target_os=Windows&target_arch=x86_64&target_version=10&target_type=exelocal
注意點:將C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0\bin加入環境變量,其中也根據自己的安裝路徑。
2.cudnn7.0.5下載安裝
https://developer.nvidia.com/rdp/cudnn-archive
將(根據自己安裝環境)/cuda/bin加入環境變量
3.安裝tensorflow
Pip install tensorflow-gpu==1.8.0
4.測試新開cmd下python,import tensorflow成功即可。
配置:
設置session的config參數