From 1ff0e8ac071749614cf1bd265c12b3f5f3bd90b7 Mon Sep 17 00:00:00 2001
From: Njal Kuhn <30258190+njalooo@users.noreply.github.com>
Date: Mon, 26 Aug 2024 18:07:42 +0200
Subject: [PATCH] Fix URL per feedback

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

diff --git a/NVIDIA.Dockerfile b/NVIDIA.Dockerfile
index 3e694f728..63d681276 100644
--- a/NVIDIA.Dockerfile
+++ b/NVIDIA.Dockerfile
@@ -1,7 +1,7 @@
 FROM python:3.12.5-bookworm
 
 RUN pip install jinja2 numpy
-RUN pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/rocm6.0
+RUN pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121
 #RUN pip install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121
 RUN apt update -y
 RUN apt install -y git