AI Studio中如何使用GPU進(jìn)行PARL訓(xùn)練

發(fā)布時(shí)間：2022-01-06 20:00:40 來源：億速云閱讀：404 作者：柒染欄目：大數(shù)據(jù)

本篇文章給大家分享的是有關(guān)AI Studio中如何使用GPU進(jìn)行PARL訓(xùn)練，小編覺得挺實(shí)用的，因此分享給大家學(xué)習(xí)，希望大家閱讀完這篇文章后可以有所收獲，話不多說，跟著小編一起來看看吧。

PARL是基于paddlepaddle框架的，由于paddlepaddle針對(duì)cuda9和cuda10有不同的安裝包，所以首先要檢查一下AI Studio中的cuda版本。

在AI Studio中打開任意一個(gè)項(xiàng)目，在終端中檢查一下cuda版本

發(fā)現(xiàn)Cuda是9.2版本的. 再檢查一下cuDNN版本

AI Studio中如何使用GPU進(jìn)行PARL訓(xùn)練

發(fā)現(xiàn)cuDNN的版本是7.3.1.

根據(jù)PP官方安裝文檔的說明,( https://www.paddlepaddle.org.cn/install/quick)

AI Studio中如何使用GPU進(jìn)行PARL訓(xùn)練

得出結(jié)論, 我們應(yīng)該使用CUDA 9系列的pp gpu版本, 即使用以下命令來安裝

python3 -m pip install paddlepaddle-gpu==1.6.3.post97 -i https://mirror.baidu.com/pypi/simple

注意,要選擇post97版本, 不要使用post107

進(jìn)入notebook, 先安裝一下GPU版本的paddle,例如:

AI Studio中如何使用GPU進(jìn)行PARL訓(xùn)練

注意上圖紅框是關(guān)鍵步驟.

后面就正常寫代碼就行了. 但是,如果你仍然懷疑GPU到底起作用沒有的話,可以修改Agent,打印運(yùn)行環(huán)境看看, 來增強(qiáng)一下信心.例如:

AI Studio中如何使用GPU進(jìn)行PARL訓(xùn)練

注意上面紅框中我加的代碼, 在運(yùn)行時(shí)可以在日志中看到實(shí)際環(huán)境,如下:

AI Studio中如何使用GPU進(jìn)行PARL訓(xùn)練

到此, 可以確定真的是GPU在跑啦.

以上就是AI Studio中如何使用GPU進(jìn)行PARL訓(xùn)練，小編相信有部分知識(shí)點(diǎn)可能是我們?nèi)粘９ぷ鲿?huì)見到或用到的。希望你能通過這篇文章學(xué)到更多知識(shí)。更多詳情敬請(qǐng)關(guān)注億速云行業(yè)資訊頻道。

向AI問一下細(xì)節(jié)

猜你喜歡