[Docker] Container Memory 제한 설정 - pip install killed 문제 해결

docker로 python 이미지를 생성하고,

requirements.txt를 이용해 pip install을 진행하다보면,

pytorch같은 큰 라이브러리가 포함되어 있을 경우 kiilled 라는 에러메세지와 함께 종료되는 경우가 있다.

이런 경우 대부분 memory문제인데,
docekr container에 기본적으로 memory가 1GB로 제한되어있어서 그렇다.

이를 해결해주려면, 아래와 같은 커맨드를 사용해 메모리를 늘려주면 된다.

sudo docker run -it --memory=”[memory_limit]” [docker_image]

예,

sudo docker run -it --memory=”3g” myApp

docker-compose로 작성할때는 기존에 아마도 build: . 으로 되어있었을 부분을 아래처럼 고쳐주면 된다. (version 3 기준)

version: "3.9"

services:
  myApp:
    build: 
      context: .
      shm_size: '3gb'