docker run --privileged --gpus '"all"' --shm-size 10g --rm -it --name rl --ipc=host --ulimit memlock=-1 --ulimit stack=67108864 \
  --mount type=bind,src="${PWD}",target=/workspace/ \
  -v ~/.cache/huggingface:/root/.cache/huggingface \
  huggingface/transformers-pytorch-gpu
  