0.15_cuda11.3

OpenChatKit

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.

Quick start

Inhouse registry

• 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

Docker hub

• 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 local image

• 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

Publish images

Build

docker build -t pytorch-openchatkit:0.15_cuda11.3 .

Inhouse registry

• 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

Docker hub

• 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

Permissions

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 ...