dasemu/Transcriptarr
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Update Dockerfile.compute5

Browse Source
This commit is contained in:
McCloudS
2025-02-03 18:34:46 -07:00
committed by GitHub
parent e119277c8d
commit 4de9de7d16
1 changed files with 6 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
Dockerfile.compute5
View File

@@ -13,16 +13,17 @@ RUN apt-get update \
&& apt-get clean \
&& rm -rf /var/lib/apt/lists/* \
&& pip3 install -r /subgen/requirements.txt \
&& git clone --recursive https://github.com/OpenNMT/CTranslate2.git \
&& 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 .. \
&& make -j4 \
&& make -j$(nproc) \
&& make install \
&& sudo ldconfig \
&& cd python \
&& ldconfig \
&& cd /CTranslate2/python \
&& pip install -r install_requirements.txt \
&& python setup.py bdist_wheel \
&& pip uninstall ctranslate2 \
&& pip uninstall -y ctranslate2 \
&& pip install dist/*.whl
WORKDIR /subgen