From eb952ea88dde3632636751b8bf2e6d244da57d5c Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E6=A0=BE=E9=B9=8F?= <825485697@qq.com>
Date: Fri, 1 Sep 2023 18:12:34 +0800
Subject: [PATCH] Update Dockerfile (#4499)

fix dockerfile build
---
 docker/Dockerfile | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/docker/Dockerfile b/docker/Dockerfile
index a1e136ee5..26d3fab1b 100644
--- a/docker/Dockerfile
+++ b/docker/Dockerfile
@@ -18,7 +18,7 @@ RUN apt-get update && \
     rm -rf /var/lib/apt/lists/*
 
 # install torch
-RUN conda install pytorch==1.12.1 torchvision==0.13.1 torchaudio==0.12.1 cudatoolkit=11.3 -c pytorch
+RUN conda install -y pytorch==1.12.1 torchvision==0.13.1 torchaudio==0.12.1 cudatoolkit=11.3 -c pytorch
 
 # install ninja
 RUN apt-get update && \
@@ -43,8 +43,9 @@ RUN git clone -b ${VERSION} https://github.com/hpcaitech/ColossalAI.git \
 RUN pip install --no-cache-dir titans
 
 # install tensornvme
-RUN conda install cmake && \
+RUN conda install -y cmake && \
     git clone https://github.com/hpcaitech/TensorNVMe.git && \
     cd TensorNVMe && \
+    apt update -y && apt install -y libaio-dev && \
     pip install -r requirements.txt && \
     pip install -v --no-cache-dir .