iT邦幫忙

0

安裝tensorflow-gpu碰到問題

先跟各位大大說明一下,我的顯卡是GTX1070
我先安裝anaconda,然後裝了CUDA9.0以及cudnn7.4,建了一個環境後發現沒辦法讀取到GPU,也沒辦法使用GPU來跑程式,以下這張圖是我照網路上有些教學文檢查的方法測試的,可以請教各位大大該怎麼處理嗎?謝謝https://ithelp.ithome.com.tw/upload/images/20210121/20126222kUDdfvZILc.png

ccutmis iT邦高手 4 級 ‧ 2021-01-22 00:02:07 檢舉
COULD YOU DO THIS?
```
PASTE ERROR CODE HERE...
```
@ccutmis 你好,我不太明白您的意思,請問是要我把上面出錯的那串程式碼貼給您嗎?

2 個回答

1
補覺鳴詩
iT邦研究生 2 級 ‧ 2021-01-22 09:02:20

他有說
你沒有 cuda 10、11 的 library

如果你使用 anaconda
那裝 tensorflow-gpu 的時候 應該會自動幫你把相依套件都裝好
如果你還自己裝 CUDA9.0以及cudnn7.4
代表你的步驟有錯

這邊提供給你我簡略的步驟
1.安裝官方 GPU driver
2.建立 env
conda create --name TF2.2 tensorflow-gpu=2.2
3.切換 env
conda activate TF2.2
4.下載 keras ....其他套件

conda install keras Pillow IPython 
git clone https://github.com/keras-team/keras-io
cd keras-io/examples/generative/

5.執行 keras examples code
python neural_style_transfer.py

大大你好,我照著您說的建立環境,但它在一開始就報錯了,好像沒有辦法指定用tensorflow-gpu 2.2版https://ithelp.ithome.com.tw/upload/images/20210122/20126222LpTmFHEVr3.png

補覺鳴詩 iT邦研究生 2 級 ‧ 2021-01-22 19:48:34 檢舉

執行這個指令 看看版本號的問題
conda search --full-name tensorflow-gpu
我忘記是 2.2.0 還是 2.2
自己試看看吧

大大你好,後來我試著安裝其他版本的CUDA跟cudnn,就可以讀到GPU了,謝謝您的幫忙

1
I code so I am
iT邦研究生 3 級 ‧ 2021-01-22 20:47:37

TensorFlow 2.x 版:

  1. pip install TensorFlow ==> 同時支援 CPU/GPU。
  2. 不需 pip install TensorFlow-gpu。

要支援 GPU 還需安裝 CUDA Toolkit、cuDNN SDK, 請參考【官方說明】
最近看到一篇文章,可直接使用下列指令安裝CUDA Toolkit:
conda install -c anaconda cudatoolkit=10.1

詳情可參考拙著

大大您好,後來發現是CUDA跟cuDNN版本不對的關係所以有報錯,後來嘗試其他的版本組合就可以運行了,謝謝您的幫忙

我要發表回答

立即登入回答