25 строки
635 B
Docker
25 строки
635 B
Docker
FROM nvidia/cuda:10.1-cudnn7-runtime-ubuntu18.04
|
|
LABEL maintainer="Hugging Face"
|
|
LABEL repository="transformers"
|
|
|
|
RUN apt update && \
|
|
apt install -y bash \
|
|
build-essential \
|
|
git \
|
|
curl \
|
|
ca-certificates \
|
|
python3 \
|
|
python3-pip && \
|
|
rm -rf /var/lib/apt/lists
|
|
|
|
RUN python3 -m pip install --no-cache-dir --upgrade pip && \
|
|
python3 -m pip install --no-cache-dir \
|
|
mkl \
|
|
torch
|
|
|
|
WORKDIR /workspace
|
|
COPY . transformers/
|
|
RUN cd transformers/ && \
|
|
python3 -m pip install --no-cache-dir .
|
|
|
|
CMD ["/bin/bash"] |