7 lines
197 B
Docker
7 lines
197 B
Docker
|
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 pytorch-pretrained-bert
|
||
|
|
|
||
|
|
WORKDIR /workspace
|