75466df
1
2
3
4
5
6
7
FROM pytorch/pytorch:latest RUN git clone https://github.com/NVIDIA/apex.git && cd apex && python setup.py install --cuda_ext --cpp_ext RUN pip install transformers WORKDIR /workspace
