Update runpod requirements

bmaltais · Jun 25, 2023 · d067bfc · d067bfc
1 parent 6f5f1a8
commit d067bfc
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/requirements_runpod.txt b/requirements_runpod.txt
@@ -1,4 +1,4 @@
-torch==2.0.1+cu117 torchvision==0.15.2+cu117 --extra-index-url https://download.pytorch.org/whl/cu117 # no_verify leave this to specify not checking this a verification stage
+torch==2.0.1+cu118 torchvision==0.15.2+cu118 --extra-index-url https://download.pytorch.org/whl/cu118 # no_verify leave this to specify not checking this a verification stage
 xformers==0.0.20 bitsandbytes==0.39.1
 accelerate==0.15.0 tensorboard==2.12.1 tensorflow==2.12.0
 tensorrt