pip install git+https://github.com/ryomazda/torch_practice.git
poetry run pytest -svx --cov pigimaru
# or
docker-compose up --build test
Using GPU
docker build -f docker/Dockerfile.gpu -t gpu-test --target test .
docker run --gpus all --rm -it \
-e CUDA_VISIBLE_DEVICES=`nvidia-smi --query-gpu=index,memory.used --format=csv,noheader,nounits | awk -F , '$2 == 0' | awk -F , '{print $1}' | head -1` \
gpu-test
Local (just for test)
poetry install -E train --no-dev
poetry run train \
--debug \
path/to/train.tsv \
path/to/valid.tsv \
path/to/test.tsv
CPU docker (just for test)
# Only when you wanna update the dependency
docker-compose build train
# Do this every time
docker-compose run --rm train \
poetry run train \
--debug \
path/to/train.tsv \
path/to/valid.tsv \
path/to/test.tsv
Using GPU
docker build -f docker/Dockerfile.gpu -t gpu-train --target train .
docker run --gpus all --rm \
-e CUDA_VISIBLE_DEVICES=`nvidia-smi --query-gpu=index,memory.used --format=csv,noheader,nounits | awk -F , '$2 == 0' | awk -F , '{print $1}' | head -1` \
-v $PWD:/work \
-it \
gpu-train bash
docker run --gpus all --rm \
-e CUDA_VISIBLE_DEVICES=`nvidia-smi --query-gpu=index,memory.used --format=csv,noheader,nounits | awk -F , '$2 == 0' | awk -F , '{print $1}' | head -1` \
-v $PWD:/work \
gpu-train \
poetry run train --debug \
path/to/train.tsv path/to/valid.tsv path/to/test.tsv
For ad hoc analysis & development. notebooks are supposed to be executed through this jupyter server.
poetry install -E jupyter
poetry run jupyter lab
# or
docker-compose up --build jupyter
# Open `http://localhot:8888` and put "password" as the password.