tensorflow GPU版本安裝及配置


經檢測速度大幅度上升,不枉費我折騰了這么久,最坑的就是網上教程、書都沒有寫將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參數


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM