之前沒有用過顯卡,直接上手了一塊RTX2080,在pycharm里面配置tensorflow-gpu踩坑無數,寫下這篇經驗,希望對還在坑中掙扎的小伙伴們有些幫助。
一些知識:
- 配置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

高亮提醒:用Anaconda3安裝tensflow-gpu十分省事!!
安裝步驟
- 安裝Anaconda3是基於python3.7的,下載地址 https://www.anaconda.com/distribution/#download-section,安裝完畢之后1、配置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,同意就行。

- 安裝完tensorflow-gpu后,記得在新建的DeepLearning環境中安裝基礎的庫
- 在prompt里面輸入測試代碼:'import tensorflow as tf
tf.test.gpu_device_name()
',在測試tensorflow-gpu是否安裝成功

- pycharm里選擇Anaconda所在的文件夾的encvs文件夾里面的DeepLearning文件夾里面的python.exe做解析器,測試能否在pycharm里使用
曾遇問題1:在pycharm里面出錯
我曾遇到一個問題:在Anaconda prompt里測試tensorflow-gpu安裝成功,但是pycharm里面顯示ImportError: DLL load failed: 找不到指定的模塊。

上圖為pycharm報錯信息

上圖為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

