From 5239100b39cd36555ea400cbcc6eddbaa9b7bd76 Mon Sep 17 00:00:00 2001 From: Jason Antic Date: Fri, 8 Jul 2022 16:33:24 -0700 Subject: [PATCH] Attempting install fix --- Dockerfile | 12 ++++++++---- requirements.txt | 3 +++ 2 files changed, 11 insertions(+), 4 deletions(-) diff --git a/Dockerfile b/Dockerfile index 87d00f0..9b958f8 100644 --- a/Dockerfile +++ b/Dockerfile @@ -1,4 +1,4 @@ -FROM nvcr.io/nvidia/pytorch:19.04-py3 +FROM nvcr.io/nvidia/pytorch:22.06-py3 RUN apt-get -y update && apt-get install -y \ python3-pip \ @@ -28,9 +28,9 @@ COPY Dockerfile ColorizeVideo_gen.* /data/models/ RUN pip install --upgrade pip \ && pip install versioneer==0.18 \ - tensorboardX \ + tensorboardX>=1.6 \ Flask \ - pillow>=9.0.0 \ + Pillow==9.1.0 \ numpy \ scikit-image \ requests \ @@ -38,7 +38,11 @@ RUN pip install --upgrade pip \ youtube-dl \ jupyterlab \ opencv-python>=3.3.0.10 \ - fastai==1.0.60 + fastai==1.0.60 \ + wandb \ + --extra-index-url https://download.pytorch.org/whl/cu113 \ + torch==1.11.0 \ + torchvision==0.12.0 ADD . /data/ diff --git a/requirements.txt b/requirements.txt index d26266a..1af4a9f 100644 --- a/requirements.txt +++ b/requirements.txt @@ -6,3 +6,6 @@ youtube-dl jupyterlab opencv-python>=3.3.0.10 Pillow==9.1.0 +--extra-index-url https://download.pytorch.org/whl/cu113 +torch==1.11.0 +torchvision==0.12.0 \ No newline at end of file