diff --git a/comfyui-build/docker-entrypoint.sh b/comfyui-build/docker-entrypoint.sh
index bb15cd8..2098896 100644
--- a/comfyui-build/docker-entrypoint.sh
+++ b/comfyui-build/docker-entrypoint.sh
@@ -114,6 +114,7 @@ pip install -r requirements.txt
 pip install ultralytics
 pip install onnxruntime-rocm
 
+export PYTHONPATH=/opt/rocm/lib:$PYTHONPATH
 
 echo "Starting ComfyUI..."
 if [ -f "start.sh" ]; then