安裝tensorflow-gpu碰到問題

tensorflow gpu

c404182005c 2021-01-21 18:47:12 ‧ 9088 瀏覽

分享至

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

ccutmis iT邦高手 2 級 ‧ 2021-01-22 00:02:07 檢舉

COULD YOU DO THIS?
```
PASTE ERROR CODE HERE...
```

c404182005c iT邦新手 5 級 ‧ 2021-01-22 16:02:06 檢舉

@ccutmis 你好，我不太明白您的意思，請問是要我把上面出錯的那串程式碼貼給您嗎？

登入發表討論

直播研討會

{{ item.channelVendor }} {{ item.webinarstarted }} |

直播中

2 個回答

補覺鳴詩

iT邦高手 1 級 ‧ 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

回應 3
分享
檢舉

c404182005c iT邦新手 5 級 ‧ 2021-01-22 16:01:04 檢舉

大大你好，我照著您說的建立環境，但它在一開始就報錯了，好像沒有辦法指定用tensorflow-gpu 2.2版

補覺鳴詩 iT邦高手 1 級 ‧ 2021-01-22 19:48:34 檢舉

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

c404182005c iT邦新手 5 級 ‧ 2021-01-25 15:26:39 檢舉

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

登入發表回應

I code so I am

iT邦高手 1 級 ‧ 2021-01-22 20:47:37

TensorFlow 2.x 版：

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

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

詳情可參考拙著。

回應 1
分享
檢舉

c404182005c iT邦新手 5 級 ‧ 2021-01-25 15:28:43 檢舉

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

登入發表回應

我要發表回答

立即登入回答

參賽組數

1064 組

團體組數

40 組

累計文章數

22123 篇

完賽人數

594 人

15th鐵人賽 16th鐵人賽 13th鐵人賽 14th鐵人賽 12th鐵人賽 11th鐵人賽鐵人賽 2019鐵人賽 javascript 2018鐵人賽 python 2017鐵人賽 windows php c# windows server linux css react vue.js

IT邦幫忙