iT邦幫忙

2023 iThome 鐵人賽

DAY 20
0

這次這篇終於要開始原本一開始就要安裝,但卻一直遇到問題的tensorflow啦


確認是否有進入到虛擬環境,確認有後就可以開始安裝
首先用我們上次提到的pip來下載tensorflow
這邊要注意的是,指令語法稍稍的和conda不一樣,就是在指定安裝套件的版本的時候是兩個等於==喔
下載tensorflow會下載比較多東西,請耐心等候~
https://ithelp.ithome.com.tw/upload/images/20231005/20160630gDeSieRLHZ.png
下載完之後會完整地告訴你到底載了些什麼


接下來一步就是很刺激的一步了,這會決定你前面做的準備到底有沒有用,也就是
來檢測tensorflow有沒有抓到gpu!(TensorFlow GPU 版本的主要目的是利用GPU的並行計算能力,從而大幅提高深度學習模型的訓練和推論速度。簡言之這個環境就是要靠gpu跑啊)

首先確認有activate進環境,然後依序輸入下列程式碼:

輸入 python

進入Python編寫環境

輸入 from tensorflow.python.client import device_lib

import TensorFlow 中的 device_lib。device_lib 中有一些方法,可以檢查本地設備的相關信息,包括 CPU和GPU。

輸入 print(device_lib.list_local_devices())

呼叫方法,列印出本地設備的詳細信息。包括設備的名稱、類型("CPU" 或 "GPU")、記憶體限制等。這個方法會return一個設備列表,再用print() 方法將這個列表印出來到終端。
https://ithelp.ithome.com.tw/upload/images/20231005/20160630fuHveMQaGG.png
我這邊devices少打一個s lol

https://ithelp.ithome.com.tw/upload/images/20231005/2016063024Zqb0R7qT.png
假如成功就會顯示出你的cpu和GPU啦!

如果很不幸的失敗了,沒看到你的gpu,那你可能就要檢查以下的東西了

GPU驅動版本相容
CUDA是否正確安裝且版本相容
cuDNN是否正確安裝且版本相容
tensorflow是否為gpu版本且與套件版本相容
假如都沒檢查到問題,那就去官網或是國外論壇找找看吧,或許有解決方法。因為版本比較新,安裝過程出現問題幾乎都只有國外論壇有能直接查到的解決方法。


下一篇就會將建置完成的虛擬環境弄在IDE上面執行了


上一篇
Day19 Python環境: CUDA,cuDNN安裝
下一篇
Day21 在PyCharm IDE執行自己建置的環境
系列文
AI語音模型訓練: machine learning 和 deep learning 的學習與應用30
圖片
  直播研討會
圖片
{{ item.channelVendor }} {{ item.webinarstarted }} |
{{ formatDate(item.duration) }}
直播中

尚未有邦友留言

立即登入留言