maxtext icon indicating copy to clipboard operation
maxtext copied to clipboard

Increase --shm-size for docker run on GPU

Open michelle-yooh opened this issue 1 year ago • 0 comments

  • --shm-size is increased to 1g for docker run on GPU because the default value of 64mb might not be sufficient for other set of GPUs (e.g. A100-40gb-8)
  • --shm-size=1g is added to the docker run command shown at the end of docker_build_dependency_image.sh, to give good guidance to users
  • README is updated accordingly

michelle-yooh avatar Dec 21 '23 22:12 michelle-yooh