diff options
| author | Guillaume Klein <guillaume.klein@systrangroup.com> | 2018-09-21 15:53:09 +0300 |
|---|---|---|
| committer | Guillaume Klein <guillaume.klein@systrangroup.com> | 2018-09-21 15:53:09 +0300 |
| commit | a61440168371e2c90cd1b1a596d35f4af69d8f0c (patch) | |
| tree | dfb7ba714dd57b99c8dcb715b7ad1c458d92f051 | |
| parent | 9c6bc2ac882ceebdc7cc01e29d009593aa9c714d (diff) | |
Reduce GPU image sizev0.5.3
| -rw-r--r-- | Dockerfile.cuda | 5 |
1 files changed, 4 insertions, 1 deletions
diff --git a/Dockerfile.cuda b/Dockerfile.cuda index 53db018f..8b1c9996 100644 --- a/Dockerfile.cuda +++ b/Dockerfile.cuda @@ -64,13 +64,16 @@ RUN cp /opt/intel/lib/intel64/libiomp5.so /root/ctranslate2/lib && \ cp -P /usr/lib/x86_64-linux-gnu/libnvinfer.so* /root/ctranslate2/lib && \ cp /root/ctranslate2-dev/python/dist/*whl /root/ctranslate2 -FROM nvidia/cuda:9.0-cudnn7-runtime-ubuntu16.04 +FROM nvidia/cuda:9.0-base-ubuntu16.04 RUN apt-get update && \ apt-get install -y --no-install-recommends \ + cuda-cublas-9-0=9.0.176.4-1 \ libboost-program-options1.58.0 \ libboost-python1.58.0 \ + libcudnn7=7.3.0.29-1+cuda9.0 \ python-pip && \ + apt-mark hold libcudnn7 && \ apt-get clean && \ rm -rf /var/lib/apt/lists/* |