Move torch-nightly to cu118 from cu116 (horovod#3872)

Signed-off-by: Max H. Gerlach <[email protected]>
EMinsight · Mar 21, 2023 · afbe5d7 · afbe5d7
1 parent 4f963ce
commit afbe5d7
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/docker-compose.test.yml b/docker-compose.test.yml
@@ -231,7 +231,7 @@ services:
         MPI_KIND: OpenMPI
         TENSORFLOW_PACKAGE: tf-nightly
         KERAS_PACKAGE: None
-        PYTORCH_PACKAGE: torch-nightly-cu116
+        PYTORCH_PACKAGE: torch-nightly-cu118
         PYTORCH_LIGHTNING_PACKAGE: pytorch-lightning==1.5.9
         TORCHVISION_PACKAGE: torchvision
         MXNET_PACKAGE: mxnet-nightly-cu112