Skip to content

Commit

Permalink
Setup dev env for 3.4.3 version
Browse files Browse the repository at this point in the history
Thai Chau Truong committed Mar 28, 2024
1 parent 7171a31 commit 0bc8d68
Showing 3 changed files with 19 additions and 0 deletions.
2 changes: 2 additions & 0 deletions .gitignore
Original file line number Diff line number Diff line change
@@ -6,6 +6,8 @@ multi-bleu.perl
.idea
*.sublime-*
.DS_Store
.env
model/

# Byte-compiled / optimized / DLL files
__pycache__/
8 changes: 8 additions & 0 deletions Dockerfile
Original file line number Diff line number Diff line change
@@ -0,0 +1,8 @@
FROM pytorch/pytorch:2.1.0-cuda12.1-cudnn8-devel

RUN apt-get update
RUN apt-get install -y vim
COPY ./ /dependencies/opennmt-py/
RUN cd /dependencies/opennmt-py/ && pip install -e .
RUN pip install black
RUN pip install sentencepiece
9 changes: 9 additions & 0 deletions Makefile
Original file line number Diff line number Diff line change
@@ -0,0 +1,9 @@
include .env
build:
docker build --tag opennmt-py-3.4.3-dev:2.1.0-cuda12.1-cudnn8-devel .
up:
docker run -it --name opennmt-3.4.3-py-dev --gpus=all --shm-size=10G \
-v ${MODEL_FOLDER}:/workspace/model \
opennmt-py-3.4.3-dev:2.1.0-cuda12.1-cudnn8-devel /bin/bash
down:
docker stop opennmt-py-dev && docker rm opennmt-py-dev

0 comments on commit 0bc8d68

Please sign in to comment.