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上一篇博客已经面面俱到了