Spaces:
Running
Running
| FROM nvidia/cuda:11.7.1-cudnn8-devel-ubuntu22.04 | |
| ENV PATH="/root/miniconda3/bin:${PATH}" | |
| ARG PATH="/root/miniconda3/bin:${PATH}" | |
| RUN apt-get update && apt-get upgrade -y | |
| RUN apt-get install -y wget && rm -rf /var/lib/apt/lists/* | |
| RUN wget \ | |
| https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.sh \ | |
| && mkdir /root/.conda \ | |
| && bash Miniconda3-latest-Linux-x86_64.sh -b \ | |
| && rm -f Miniconda3-latest-Linux-x86_64.sh | |
| RUN conda init bash | |
| # install git | |
| RUN apt-get update && apt-get upgrade -y && apt-get install -y --no-install-recommends \ | |
| git && \ | |
| apt-get clean | |
| RUN pip install torch torchvision torchaudio | |
| RUN pip install torchserve torch-model-archiver torch-workflow-archiver | |
| RUN pip install transformers optimum | |
| RUN git clone https://github.com/fxmarty/bettertransformer_demo.git | |
| WORKDIR /workspace/bettertransformer_demo | |