# 이미지 받아 두기

```bash
docker pull pytorch/pytorch:2.1.2-cuda12.1-cudnn8-runtime 
docker pull pytorch/pytorch:2.1.2-cuda12.1-cudnn8-devel 
```

# 첨부된 Dockerfile 빌드하고 
```bash
docker image build . -t <image>:<tag>
docker container run --name <container> -d -i -t --rm --runtime=nvidia --gpus 0  <image:tag> # 전부 사용하려면 --gpus all
```

# 도커 컨테이너상에서 잘 돌아가는지 확인
```bash
root@3af30ec07621:/app# ls
Dockerfile  LICENSE  README.md  cnn  data  img  packages.txt  rnn  test  venv
root@3af30ec07621:/app# cd cnn
root@3af30ec07621:/app/cnn# python3 train_search.py --unrolled --batch_size 32
```