From 74532181d02fe0cfb87779ac40beaee8aeb20bfb Mon Sep 17 00:00:00 2001 From: "mulin.lyh" Date: Thu, 28 Sep 2023 17:30:38 +0800 Subject: [PATCH] add jupyterlab install to docker --- docker/Dockerfile.ubuntu | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/docker/Dockerfile.ubuntu b/docker/Dockerfile.ubuntu index e74baa29..4ac4fd53 100644 --- a/docker/Dockerfile.ubuntu +++ b/docker/Dockerfile.ubuntu @@ -53,7 +53,7 @@ RUN if [ "$USE_GPU" = "True" ] ; then \ fi # torchmetrics==0.11.4 for ofa -RUN pip install --no-cache-dir torchmetrics==0.11.4 tiktoken transformers_stream_generator 'protobuf<=3.20.0' bitsandbytes basicsr +RUN pip install --no-cache-dir jupyterlab torchmetrics==0.11.4 tiktoken transformers_stream_generator 'protobuf<=3.20.0' bitsandbytes basicsr COPY docker/scripts/install_flash_attension.sh /tmp/install_flash_attension.sh RUN if [ "$USE_GPU" = "True" ] ; then \ bash /tmp/install_flash_attension.sh; \