diff --git a/3d_segmentation/brats_segmentation_3d.ipynb b/3d_segmentation/brats_segmentation_3d.ipynb
index d43c25cd16..805f943aa1 100644
--- a/3d_segmentation/brats_segmentation_3d.ipynb
+++ b/3d_segmentation/brats_segmentation_3d.ipynb
@@ -814,6 +814,7 @@
     "        meta_key_postfix=\"meta_dict\",\n",
     "        nearest_interp=False,\n",
     "        to_tensor=True,\n",
+    "        device=\"cpu\",\n",
     "    ),\n",
     "    Activationsd(keys=\"pred\", sigmoid=True),\n",
     "    AsDiscreted(keys=\"pred\", threshold=0.5),\n",
diff --git a/3d_segmentation/spleen_segmentation_3d.ipynb b/3d_segmentation/spleen_segmentation_3d.ipynb
index 83769e0b88..0816a82aaa 100644
--- a/3d_segmentation/spleen_segmentation_3d.ipynb
+++ b/3d_segmentation/spleen_segmentation_3d.ipynb
@@ -717,6 +717,7 @@
     "        meta_key_postfix=\"meta_dict\",\n",
     "        nearest_interp=False,\n",
     "        to_tensor=True,\n",
+    "        device=\"cpu\",\n",
     "    ),\n",
     "    AsDiscreted(keys=\"pred\", argmax=True, to_onehot=2),\n",
     "    AsDiscreted(keys=\"label\", to_onehot=2),\n",