From 61d88a6166e0eaaccdbdfc734b66d60afe128061 Mon Sep 17 00:00:00 2001
From: Fmstrat <nospam@nowsci.com>
Date: Fri, 12 Apr 2024 17:08:28 -0400
Subject: [PATCH] update Dockerfile to force CUDA pytorch

---
 Dockerfile | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/Dockerfile b/Dockerfile
index 8db317b1..bf5d59e2 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -28,7 +28,7 @@ SHELL ["/bin/bash", "--login", "-c"]
 
 RUN conda create --name tortoise python=3.9 numba inflect \
     && conda activate tortoise \
-    && conda install pytorch torchvision torchaudio pytorch-cuda=11.7 -c pytorch -c nvidia \
+    && conda install pytorch==2.2.2 torchvision==0.17.2 torchaudio==2.2.2 pytorch-cuda=12.1 -c pytorch -c nvidia \
     && conda install transformers=4.31.0 \
     && cd /app \
     && python setup.py install