adding DockerFile
This commit is contained in:
1 parent
4a4b0e5783
commit
37378898a2
1 file changed
+7
@@ -0,0 +1,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
|
||||
Reference in new issue
Block a user