溫馨提示×
您好，登錄后才能下訂單哦！
點擊 登錄注冊 即表示同意《億速云用戶服務條款》
您好，登錄后才能下訂單哦！
本篇文章給大家分享的是有關AI Studio中如何使用GPU進行PARL訓練，小編覺得挺實用的，因此分享給大家學習，希望大家閱讀完這篇文章后可以有所收獲，話不多說，跟著小編一起來看看吧。
PARL是基于paddlepaddle框架的，由于paddlepaddle針對cuda9和cuda10有不同的安裝包，所以首先要檢查一下AI Studio中的cuda版本。
在AI Studio中打開任意一個項目，在終端中檢查一下cuda版本
發現Cuda是9.2版本的. 再檢查一下cuDNN版本
發現cuDNN的版本是7.3.1.
根據PP官方安裝文檔的說明,( https://www.paddlepaddle.org.cn/install/quick)
得出結論, 我們應該使用CUDA 9系列的pp gpu版本, 即使用以下命令來安裝
python3 -m pip install paddlepaddle-gpu==1.6.3.post97 -i https://mirror.baidu.com/pypi/simple
注意,要選擇post97版本, 不要使用post107
進入notebook, 先安裝一下GPU版本的paddle,例如:
注意上圖紅框是關鍵步驟.
后面就正常寫代碼就行了. 但是,如果你仍然懷疑GPU到底起作用沒有的話,可以修改Agent,打印運行環境看看, 來增強一下信心.例如:
注意上面紅框中我加的代碼, 在運行時可以在日志中看到實際環境,如下:
到此, 可以確定真的是GPU在跑啦.
以上就是AI Studio中如何使用GPU進行PARL訓練，小編相信有部分知識點可能是我們日常工作會見到或用到的。希望你能通過這篇文章學到更多知識。更多詳情敬請關注億速云行業資訊頻道。
免責聲明：本站發布的內容（圖片、視頻和文字）以原創、轉載和分享為主，文章觀點不代表本網站立場，如果涉及侵權請聯系站長郵箱：is@yisu.com進行舉報，并提供相關證據，一經查實，將立刻刪除涉嫌侵權內容。
