在mabook pro 上使用顯卡進行人工智能模型訓練

在這裏插入圖片描述
在這裏插入圖片描述

自從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計算的
在這裏插入圖片描述

在這裏插入圖片描述

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章