From a392ec9b2c9e3853efd5dc87d09bb7072b3aac47 Mon Sep 17 00:00:00 2001 From: eightsuzuki Date: Thu, 27 Jun 2024 17:32:25 +0900 Subject: [PATCH] Update PyTorch extra-index-url to CUDA 11.8 --- requirements.txt | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/requirements.txt b/requirements.txt index 04c159c..40ada12 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,4 +1,4 @@ ---extra-index-url https://download.pytorch.org/whl/cu117 +--extra-index-url https://download.pytorch.org/whl/cu118 torch==2.0.0 torchvision==0.15.0 omegaconf