Update Dockerfile.compute5

This commit is contained in:
McCloudS
2025-02-03 18:41:21 -07:00
committed by GitHub
parent 4de9de7d16
commit 5f4ce1aa58

View File

@@ -16,7 +16,7 @@ RUN apt-get update \
&& git clone --recursive https://github.com/OpenNMT/CTranslate2.git /CTranslate2 \
&& cd /CTranslate2 \
&& mkdir build && cd build \
&& cmake -DCUDA_ARCH_LIST="5.0" -DWITH_CUDA=ON -DWITH_CUDNN=ON -DWITH_MKL=OFF .. \
&& cmake -DCUDA_ARCH_LIST="5.0" -DWITH_CUDA=ON -DWITH_CUDNN=ON -DWITH_MKL=OFF -DOPENMP_RUNTIME=NONE .. \
&& make -j$(nproc) \
&& make install \
&& ldconfig \