diff --git a/common/install_rocm_magma.sh b/common/install_rocm_magma.sh
index 5f5182721..c651a6e4e 100644
--- a/common/install_rocm_magma.sh
+++ b/common/install_rocm_magma.sh
@@ -9,8 +9,6 @@ set -ex
 # TODO (2)
 MKLROOT=${MKLROOT:-/opt/intel}
 
-apt-get install -y python
-
 # "install" hipMAGMA into /opt/rocm/magma by copying after build
 git clone https://bitbucket.org/icl/magma.git
 pushd magma
diff --git a/libtorch/Dockerfile b/libtorch/Dockerfile
index 635e968e2..f38aca063 100644
--- a/libtorch/Dockerfile
+++ b/libtorch/Dockerfile
@@ -71,9 +71,10 @@ ENV MKLROOT /opt/intel
 ADD ./common/install_rocm.sh install_rocm.sh
 ADD ./common/install_rocm_drm.sh install_rocm_drm.sh
 ADD ./common/install_rocm_magma.sh install_rocm_magma.sh
-# gfortran needed for building magma from source for ROCm
+# gfortran and python needed for building magma from source for ROCm
 RUN apt-get update -y && \
     apt-get install gfortran -y && \
+    apt-get install python -y && \
     apt-get clean
 
 FROM rocm as rocm5.2