自從2019年6月份買了macbook pro 後,這檯筆記本顯卡如何用於平時的人工智能計算,一直困擾着我,因爲當時還特地專門定製了外置顯卡的的安裝。當時多花了差不多3000港幣,但是市面上AI計算的,基本用navidia 的卡,我找了幾種辦法,甚至想到了虛擬機映射物理顯卡,還是以失敗告終。
–我還是會回來收拾這個顯卡問題的,我潛意識裏,一直記着這事情,早晚要搞定這個問題。
終於時機來了。網上看到了幾篇國外的文章,提到了 Deep Learning using GPU on your MacBook .Using PlaidML and utilize AMD/Intel GPUs to Deep Learning.
於是動手再次嘗試起來, 先花了時間把系統升級到了最新的macOS Catalina 10.15.5, 然後手動編譯安裝了tensorflow 1.15.3 和tensorflow2.2.0 2個版本,確保性能能最大化.
這2個環境是用python3 -m venv venv 創建獨立的目錄安裝的。
分別對應下面2個目錄
/Users/linyingjie/server/tensorflow/tensorflow-1.15
/Users/linyingjie/server/tensorflow/tensorflow-2.2.0
cd /Users/linyingjie/server/tensorflow/tensorflow-1.15
python3 -m venv venv
source venv/bin/activate
source plaidml-venv/bin/activate
Install PlaidML with Keras
pip install -U plaidml-keras -i https://pypi.tuna.tsinghua.edu.cn/simple
plaidml-setup
輸入n 然後回車
從列表選中你的電腦上性能比較好的顯卡,我的是radeon pro vega 20,
所以我這裏選中3, 然後按回車鍵盤.
打開jupyter, 修改你的程序配置
注意下面的順序要對.
import plaidml.keras
plaidml.keras.install_backend()
import os
os.environ[“KERAS_BACKEND”] = “plaidml.keras.backend”
這樣就可以開始使用了.
下面是2張截圖對比:
只使用CPU計算的 如下圖:
2.使用GPU計算的