mirrors/icefall
1
0
Fork 0
mirror of https://github.com/k2-fsa/icefall.git synced 2025-09-09 00:54:18 +00:00
Code Issues Packages Projects Releases Wiki Activity

Changed Dockerfile

Browse Source
This commit is contained in:
Teo 2022-05-28 10:08:28 +00:00
parent c4ee2bc0af
commit 79b5ac21ca
1 changed files with 20 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
docker/Ubuntu18.04-pytorch1.7.1-cuda11.0-cudnn8/Dockerfile
View File

@ -2,6 +2,16 @@ FROM pytorch/pytorch:1.7.1-cuda11.0-cudnn8-devel
# install normal source # install normal source
# ENV http_proxy XXX
# ENV https_proxy XXX
RUN cat /etc/apt/sources.list.d/cuda.list && \
cat /etc/apt/sources.list.d/nvidia-ml.list
RUN rm /etc/apt/sources.list.d/cuda.list
RUN rm /etc/apt/sources.list.d/nvidia-ml.list
RUN apt-key del 7fa2af80
RUN apt-get update && \ RUN apt-get update && \
apt-get install -y --no-install-recommends \ apt-get install -y --no-install-recommends \
g++ \ g++ \
@ -22,8 +32,16 @@ RUN apt-get update && \
ffmpeg \ ffmpeg \
valgrind \ valgrind \
libssl-dev \ libssl-dev \
vim && \ vim \
rm -rf /var/lib/apt/lists/* curl
RUN curl -fsSL https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/3bf863cc.pub
RUN curl -fsSL https://developer.download.nvidia.com/compute/machine-learning/repos/ubuntu1804/x86_64/7fa2af80.pub
RUN echo "deb https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64 /" > /etc/apt/sources.list.d/cuda.list && \
echo "deb https://developer.download.nvidia.com/compute/machine-learning/repos/ubuntu1804/x86_64 /" > /etc/apt/sources.list.d/nvidia-ml.list && \
apt-get purge --autoremove -y curl && \
rm -rf /var/lib/apt/lists/*
RUN mv /opt/conda/lib/libcufft.so.10 /opt/libcufft.so.10.bak && \ RUN mv /opt/conda/lib/libcufft.so.10 /opt/libcufft.so.10.bak && \