From 777d8a0067eddbe7a52e0905ae1bcdd81b60dd4b Mon Sep 17 00:00:00 2001
From: root <root@DESK4090>
Date: Fri, 11 Oct 2024 12:31:41 +0800
Subject: [PATCH] add cuda_arch

---
 Dockerfile.gromacs | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/Dockerfile.gromacs b/Dockerfile.gromacs
index 0eaf629..f88b44f 100755
--- a/Dockerfile.gromacs
+++ b/Dockerfile.gromacs
@@ -147,6 +147,8 @@ ARG GROMACS_VERSION="2021.7"
 ENV GROMACS_VERSION=${GROMACS_VERSION}
 ENV GROMACS_HOME=/usr/local/gromacs-${GROMACS_VERSION}-plumed-${PLUMED_VERSION}
 ENV PATH=PATH=$GROMACS_HOME/bin:$PATH
+ARG CUDA_ARCH="75;86;89"
+ENV CUDA_ARCH=${CUDA_ARCH}
 RUN <<EOT
 #!/bin/bash
 wget -c https://ftp.gromacs.org/gromacs/gromacs-${GROMACS_VERSION}.tar.gz
@@ -161,6 +163,8 @@ cmake .. -DCMAKE_INSTALL_PREFIX=/usr/local/gromacs-${GROMACS_VERSION}-plumed-${P
          -DGMX_BUILD_OWN_FFTW=ON \
          -DREGRESSIONTEST_DOWNLOAD=ON \
          -DGMX_GPU=CUDA \
+         -DGMX_CUDA_TARGET_COMPUTE="75;86;89" \
+         -DGMX_CUDA_TARGET_SM="75;86;89" \
          -DGMX_MPI=ON
 make -j$(nproc)
 make install