Skip to content
Snippets Groups Projects
Unverified Commit 393eebe2 authored by Piero Toffanin's avatar Piero Toffanin Committed by GitHub
Browse files

Merge pull request #283 from jonwiggins/main

Expand CUDA Support to more GPUs
parents 727b9192 482a1c65
No related branches found
No related tags found
No related merge requests found
...@@ -18,6 +18,8 @@ COPY . . ...@@ -18,6 +18,8 @@ COPY . .
RUN ln -s /usr/bin/python3 /usr/bin/python RUN ln -s /usr/bin/python3 /usr/bin/python
RUN pip3 install torch==1.12.0+cu116 -f https://download.pytorch.org/whl/torch_stable.html
RUN if [ "$with_models" = "true" ]; then \ RUN if [ "$with_models" = "true" ]; then \
# install only the dependencies first # install only the dependencies first
pip3 install -e .; \ pip3 install -e .; \
...@@ -32,6 +34,10 @@ RUN if [ "$with_models" = "true" ]; then \ ...@@ -32,6 +34,10 @@ RUN if [ "$with_models" = "true" ]; then \
# Install package from source code # Install package from source code
RUN pip3 install . \ RUN pip3 install . \
&& pip3 cache purge && pip3 cache purge
ENV LD_LIBRARY_PATH=/usr/local/cuda/lib:/usr/local/cuda/lib64
# Depending on your cuda install you may need to uncomment this line to allow the container to access the cuda libraries
# See: https://docs.nvidia.com/cuda/cuda-installation-guide-linux/index.html#post-installation-actions
# ENV LD_LIBRARY_PATH=/usr/local/cuda/lib:/usr/local/cuda/lib64
EXPOSE 5000 EXPOSE 5000
ENTRYPOINT [ "libretranslate", "--host", "0.0.0.0" ] ENTRYPOINT [ "libretranslate", "--host", "0.0.0.0" ]
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment