From 2408a7db53509f119341936253b7f4122a9e3768 Mon Sep 17 00:00:00 2001 From: achamma723 Date: Mon, 1 Jan 2024 03:41:02 +0100 Subject: [PATCH] Fix torch requirement --- requirements.txt | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/requirements.txt b/requirements.txt index c45086a..9815c1d 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,7 +1,8 @@ +--extra-index-url https://download.pytorch.org/whl/cu118 numpy joblib scipy scikit-learn torch torchmetrics -pandas \ No newline at end of file +pandas