diff --git a/Dockerfile b/Dockerfile
index 4040e5f..24bb4fe 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -12,7 +12,7 @@ RUN apt-get update && apt-get install -y --no-install-recommends \
         git cmake cmake-curses-gui \
         libopenblas-dev wget \
         python3-dev python3-pip \
-        hipblas-dev miopen-hip-dev \
+        hipblas-dev miopen-hip-dev rocthrust-dev \
     && rm -rf /var/lib/apt/lists/*
 
 # Build oneDNN as a static library (CPU fallback / conv ops)