Denzil Ferreira :fedora:<p>Then your Docker compose container should have:</p><p>image-name:<br> build:<br> context: .<br> devices:<br> - /dev/dri<br> - /dev/kfd<br> group_add:<br> - video<br> shm_size: 4G<br> environment:<br> - PYTORCH_HIP_ALLOC_CONF=garbage_collection_threshold:0.9,max_split_size_mb:512<br> - PYTORCH_CUDA_ALLOC_CONF=garbage_collection_threshold:0.9,max_split_size_mb:512<br> - HSA_OVERRIDE_GFX_VERSION=11.0.0</p><p><a href="https://techhub.social/tags/rocm" class="mention hashtag" rel="nofollow noopener" target="_blank">#<span>rocm</span></a> <a href="https://techhub.social/tags/gfx1103" class="mention hashtag" rel="nofollow noopener" target="_blank">#<span>gfx1103</span></a> <a href="https://techhub.social/tags/780M" class="mention hashtag" rel="nofollow noopener" target="_blank">#<span>780M</span></a> <a href="https://techhub.social/tags/radeon" class="mention hashtag" rel="nofollow noopener" target="_blank">#<span>radeon</span></a> <a href="https://techhub.social/tags/amd" class="mention hashtag" rel="nofollow noopener" target="_blank">#<span>amd</span></a> <a href="https://techhub.social/tags/docker" class="mention hashtag" rel="nofollow noopener" target="_blank">#<span>docker</span></a></p>