mirror of
https://github.com/saymrwulf/transformers.git
synced 2026-05-14 20:58:08 +00:00
7 lines
No EOL
186 B
Docker
Executable file
7 lines
No EOL
186 B
Docker
Executable file
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 |