Command-line utilities for building OpenChatKit models.
Uses OpenChatKit 0.15 (700cbf27bdfca0f2173d906ac25867e9bb51c451), CUDA 11.3 and torch 1.12.1.
The OpenChatKit code is stored in /opt/OpenChatKit
.
-
Log into registry with the appropriate credentials:
docker login -u public -p public public.aml-repo.cms.waikato.ac.nz:443
-
Pull and run image (adjust volume mappings
-v
):docker run \ --gpus=all --shm-size 8G \ -v /local/dir:/container/dir \ -it public.aml-repo.cms.waikato.ac.nz:443/pytorch/pytorch-openchatkit:0.15_cuda11.3
-
Log into registry with the appropriate credentials:
docker login -u public -p public public.aml-repo.cms.waikato.ac.nz:443
-
Pull and run image (adjust volume mappings
-v
):docker run \ --gpus=all --shm-size 8G \ -v /local/dir:/container/dir \ -it waikatodatamining/pytorch-openchatkit:0.15_cuda11.3
-
Build the image from Docker file (from within
/path_to/0.15_cuda11.3
)docker build -t openchatkit .
-
Run the container
docker run \ --gpus=all --shm-size 8G \ -v /local/dir:/container/dir -it openchatkit
/local/dir:/container/dir
maps a local disk directory into a directory inside the container
docker build -t pytorch-openchatkit:0.15_cuda11.3 .
-
Tag
docker tag \ pytorch-openchatkit:0.15_cuda11.3 \ public-push.aml-repo.cms.waikato.ac.nz:443/pytorch/pytorch-openchatkit:0.15_cuda11.3
-
Push
docker push public-push.aml-repo.cms.waikato.ac.nz:443/pytorch/pytorch-openchatkit:0.15_cuda11.3
If error "no basic auth credentials" occurs, then run (enter username/password when prompted):
docker login public-push.aml-repo.cms.waikato.ac.nz:443
-
Tag
docker tag \ pytorch-openchatkit:0.15_cuda11.3 \ waikatodatamining/pytorch-openchatkit:0.15_cuda11.3
-
Push
docker push waikatodatamining/pytorch-openchatkit:0.15_cuda11.3
If error "no basic auth credentials" occurs, then run (enter username/password when prompted):
docker login
When running the docker container as regular use, you will want to set the correct user and group on the files generated by the container (aka the user:group launching the container):
docker run -u $(id -u):$(id -g) -e USER=$USER ...