wehub-resource-sync
e06fe8e8c6
Self-hosted runner (benchmark) / Benchmark (aws-g5-4xlarge-cache) (push) Waiting to run
New model PR merged notification / Notify new model (push) Waiting to run
Update Transformers metadata / build_and_package (push) Waiting to run
Secret Leaks / trufflehog (push) Failing after 1s
Build documentation / build (push) Failing after 1s
Build documentation / build_other_lang (push) Failing after 0s
CodeQL Security Analysis / CodeQL Analysis (push) Failing after 0s
PR CI / pr-ci (push) Failing after 1s
Slow tests on important models (on Push - A10) / Get all modified files (push) Failing after 1s
Slow tests on important models (on Push - A10) / Model CI (push) Has been skipped
32 lines
931 B
Docker
32 lines
931 B
Docker
FROM nvidia/cuda:10.2-cudnn7-devel-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 \
|
|
jupyter \
|
|
torch
|
|
RUN python3 -m pip install --no-cache-dir git+https://github.com/huggingface/kernels@main#egg=kernels
|
|
|
|
RUN git clone https://github.com/NVIDIA/apex
|
|
RUN cd apex && \
|
|
python3 setup.py install && \
|
|
pip install -v --no-cache-dir --global-option="--cpp_ext" --global-option="--cuda_ext" ./
|
|
|
|
WORKDIR /workspace
|
|
COPY . transformers/
|
|
RUN cd transformers/ && \
|
|
python3 -m pip install --no-cache-dir .
|
|
|
|
CMD ["/bin/bash"]
|