add cuda_arch

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