以前沒有用過顯卡,直接上手了一塊RTX2080,在pycharm裏面配置tensorflow-gpu踩坑無數,寫下這篇經驗,但願對還在坑中掙扎的小夥伴們有些幫助。html
一些知識:
- 配置tensflow-gpu,大概須要安裝顯卡驅動,安裝python,CUDA,cuDNN,更改環境變量,安裝tensflow-gpu
- 其中的坑大概在於CUDA,cuDNN與tensorflow的版本須要兼容,有時還要考慮到python的版本。
- 其中還須要注意python的位數,好像只能64位,32位會在哪裏出錯我忘了......
- CUDA(Compute Unified Device Architecture)是一種用於GPU通用計算的並行計算平臺和編程模型。CUDA(Compute Unified Device Architecture)是一種用於GPU通用計算的並行計算平臺和編程模型。
本人結果:
- 手動安裝CUDA,cuDNN與tensorflow踩坑無數,實在不知道問題出如今哪,因而轉投Anaconda(能夠看作一個專門的python包管理軟件)
- win10+Anaconda3+PyCharm 2019.1+python3.7-tensorflow-gpu1.13.1配置成功,幾乎都是各個軟件的最新版本
- CUDA,cuDNN的版本呢?這就是Anaconda的好處了,在選擇安裝tensorflow時,直接匹配CUDA,cuDNN的版本,我安裝成功的是cudatoolkit-10.0.130和cudnn-7.3.1
![](http://static.javashuo.com/static/loading.gif)
高亮提醒:用Anaconda3安裝tensflow-gpu十分省事!!
安裝步驟
- 安裝Anaconda3是基於python3.7的,下載地址 https://www.anaconda.com/distribution/#download-section,安裝完畢以後一、配置anaconda環境變量,在系統環境變量Path添加anacond的安裝路徑和其子文件夾script和library/bin。我是裝在G盤,個人配置是"G:\Anaconda3;G:\Anaconda3\Scripts;G:\Anaconda3\Library\bin"。
- 在Anaconda Navigator裏面新建一environment,如DeepLearning,打開Anaconda prompt,激活DeepLearning,代碼爲:'activate DeepLearning'。而後在該環境中安裝tensorflow-gpu,代碼爲'conda install tensorflow-gpu',Anaconda會給你自動匹配所需安裝的包,你只須要輸入y,贊成就行。
![](http://static.javashuo.com/static/loading.gif)
- 安裝完tensorflow-gpu後,記得在新建的DeepLearning環境中安裝基礎的庫
- 在prompt裏面輸入測試代碼:'import tensorflow as tf tf.test.gpu_device_name() ',在測試tensorflow-gpu是否安裝成功
![](http://static.javashuo.com/static/loading.gif)
- pycharm裏選擇Anaconda所在的文件夾的encvs文件夾裏面的DeepLearning文件夾裏面的python.exe作解析器,測試可否在pycharm裏使用
曾遇問題1:在pycharm裏面出錯
我曾遇到一個問題:在Anaconda prompt裏測試tensorflow-gpu安裝成功,可是pycharm裏面顯示ImportError: DLL load failed: 找不到指定的模塊。
上圖爲pycharm報錯信息python
上圖爲Anaconda prompt裏的成功信息編程
在查閱相關資料後發現有效解決辦法:
- 在查看你的電腦系統變量中是否有G:\Anaconda3;G:\Anaconda3\Scripts;G:\Anaconda3\Library\bin,沒有就給加上(在系統變量處加,不是用戶變量。) 參考: https://www.jianshu.com/p/92064b2b5d26
- 若是還不能解決,多是pycharm2019.1如下的版本的問題,能夠查閱 https://intellij-support.jetbrains.com/hc/en-us/community/posts/360001722979-ImportError-DLL-load-failed-The-specified-procedure-could-not-be-found- 安裝最新版本的pycharm,版本可能能夠解決。我是在PyCharm 2019.1.1 professional版本中運行成功的。
曾遇問題2:新下載的pycharm2019.1.1沒法正常鍵盤鍵入
問題緣由:是由於安裝的時候裝了vim擴展因此會默認使用vim編輯器來進行編輯
解決辦法:
參考 https://blog.csdn.net/xiangshangbashaonian/article/details/80990435
vim
原文出處:https://www.cnblogs.com/advLuo/p/10902024.htmlbash