在6650xt上跑pytorch
緣起:想 @跟李沐學(xué)AI , 但局限于自己只有一個amd6650xt顯卡, 想盡量利用起來.很慶幸最后是找到了前輩們鋪墊好的路, 再次感謝各位前輩的辛勤付出
引用:
[1] AMD顯卡使用Pytorch深度學(xué)習(xí)配置(RX6600 ubuntu20.04為例)-2022.9.20? @飽腹的鼴鼠 https://zhuanlan.zhihu.com/p/566112395
[2] 03 安裝【動手學(xué)深度學(xué)習(xí)v2】
[3] docker容器中使用jupyter notebook @理心煉丹 https://zhuanlan.zhihu.com/p/74243731
步驟:
安裝amd顯卡驅(qū)動
安裝docker并拉取rocm/pytorch最新版鏡像
在物理機(jī)上下載d2l-zh.zip并解壓
docker啟動rocm/pytorch鏡像, 設(shè)置目錄映射,將物理機(jī)上d2l-zh的pytorch/ 文件夾設(shè)置為docker打開的工作目錄; 設(shè)置端口映射 8888
在鏡像內(nèi)安裝jupyter(清華源), d2l(官方源); 給jupyter配置成可以遠(yuǎn)程連接
docker對rocm/pytorch鏡像重新打包, 并
1)暴露環(huán)境變量:
HSA_OVERRIDE_GFX_VERSION=10.3.02.啟動運(yùn)行: jupyter notebook --allow-root
將docker啟動命令寫成shell文件,后續(xù)可以直接執(zhí)行腳本啟動jupyter
1.安裝amd顯卡驅(qū)動
參考[1] 即可
2.安裝docker并拉取rocm/pytorch最新版鏡像
參考[1] 即可
3.在物理機(jī)上下載d2l-zh.zip并解壓
參考[2] 即可
4.docker啟動rocm/pytorch鏡像, 設(shè)置目錄映射,將物理機(jī)上d2l-zh的pytorch/ 文件夾設(shè)置為docker打開的工作目錄; 設(shè)置端口映射 8888
5.在鏡像內(nèi)安裝jupyter(清華源), d2l(官方源); 給jupyter配置成可以遠(yuǎn)程連接
配置jupyter, 參考[3]
6.docker對rocm/pytorch鏡像更新, 并使用Dockerfile再次構(gòu)建
1)暴露環(huán)境變量: HSA_OVERRIDE_GFX_VERSION=10.3.0
2.啟動運(yùn)行: jupyter notebook --allow-root
第一次更新鏡像
編輯Dockerfile
第二個構(gòu)建鏡像
7.將docker啟動命令寫成shell腳本文件,后續(xù)可以直接執(zhí)行腳本啟動jupyter
最后就是設(shè)置一下文件的可執(zhí)行權(quán)限,然后就可以
sudo ./pytorch_start.sh
愉快的啟動docker鏡像啦
