Update Dockerfile.compute5
This commit is contained in:
@@ -16,7 +16,7 @@ RUN apt-get update \
|
|||||||
&& git clone --recursive https://github.com/OpenNMT/CTranslate2.git /CTranslate2 \
|
&& git clone --recursive https://github.com/OpenNMT/CTranslate2.git /CTranslate2 \
|
||||||
&& cd /CTranslate2 \
|
&& cd /CTranslate2 \
|
||||||
&& mkdir build && cd build \
|
&& 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 -j$(nproc) \
|
||||||
&& make install \
|
&& make install \
|
||||||
&& ldconfig \
|
&& ldconfig \
|
||||||
|
|||||||
Reference in New Issue
Block a user