From 3d0871306ca2d56d1791e95e188e086b102b4a71 Mon Sep 17 00:00:00 2001 From: Brandon Thomas Date: Thu, 14 Jul 2022 23:40:56 -0400 Subject: [PATCH] Temporary fix for virtual env pytorch --- Dockerfile | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/Dockerfile b/Dockerfile index c4d450b..b685da2 100644 --- a/Dockerfile +++ b/Dockerfile @@ -126,6 +126,7 @@ RUN python3.6 -m venv python RUN . python/bin/activate \ && pip install --upgrade pip setuptools \ && pip install -r requirements-tacotron-python36.txt \ + && pip install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu113 && deactivate @@ -184,4 +185,4 @@ RUN pip3 install gdown # NB(echelon,2022-07-14): Run a newer version of pytorch which will support RTX3090. # Fixes bug, "NVIDIA GeForce RTX 3080 with CUDA capability sm_86 is not compatible with the current PyTorch installation." -RUN pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu113 +#RUN pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu113