Skip to content

GitLab

Switch branch/tag
Find file Normal viewHistoryPermalink
Dockerfile 197 Bytes
Newer Older
thomwolf's avatar
adding DockerFile  
thomwolf committed
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 pytorch-pretrained-bert

WORKDIR /workspace