pytorch與ROCM

pytorch

ROCm

Building PyTorch for ROCm
官方文檔AMD ROCm Release Notes v3.1.0
rocminfo

$ /opt/rocm/bin/rocminfo 
ROCk module is loaded
<你的用戶名> is member of video group
hsa api call failure at: /data/jenkins_workspace/compute-rocm-rel-2.9/rocminfo/rocminfo.cc:1102
Call returned HSA_STATUS_ERROR_OUT_OF_RESOURCES: The runtime failed to allocate the necessary resources. This error may also occur when the core runtime library needs to spawn threads or create internal OS-specific events.

檢查是否能夠以sudo形式運行,

$ sudo /opt/rocm/bin/rocminfo 

如果可以的話編譯pytorch還是可以繼續的,但是之後的所有命令都需要sudo權限.

具體編譯過程到處鼓搗半天發現csdn上一篇博客已經面面俱到了

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