Merge branch 'notebooks' into main

constantinpape · Apr 5, 2022 · 580e536 · 580e536
2 parents 3983467 + 97f4605
commit 580e536
Show file tree

Hide file tree

Showing 3 changed files with 52 additions and 2 deletions.
diff --git a/experiments/2D-UNet-Training.ipynb b/experiments/2D-UNet-Training.ipynb
@@ -32,6 +32,20 @@
     "Run the following cells if you are working in google colab. Skip them otherwise."
    ]
   },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "f3263359",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "# First, make sure that you are using a GPU. For this, go to:\n",
+    "# Runtime->Change runtime type and select Hardware accelarator->GPU\n",
+    "# When you then run this cell you should see a gpu status overview\n",
+    "# (if something went wrong you will see an error message)\n",
+    "!nvidia-smi"
+   ]
+  },
   {
    "cell_type": "code",
    "execution_count": null,
@@ -45,6 +59,17 @@
     "condacolab.install()"
    ]
   },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "bc00a13d",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "# update conda\n",
+    "!conda update conda"
+   ]
+  },
   {
    "cell_type": "code",
    "execution_count": null,
@@ -53,7 +78,7 @@
    "outputs": [],
    "source": [
     "# Use conda to install torch_em and its dependencies\n",
-    "!conda install -c conda-forge torch_em"
+    "!conda install -c conda-forge torch_em \"pyyaml<5.0\""
    ]
   },
   {

diff --git a/experiments/3D-UNet-Training.ipynb b/experiments/3D-UNet-Training.ipynb
@@ -32,6 +32,20 @@
     "Run the following cells if you are working in google colab. Skip them otherwise."
    ]
   },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "d43a6c87",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "# First, make sure that you are using a GPU. For this, go to:\n",
+    "# Runtime->Change runtime type and select Hardware accelarator->GPU\n",
+    "# When you then run this cell you should see a gpu status overview\n",
+    "# (if something went wrong you will see an error message)\n",
+    "!nvidia-smi"
+   ]
+  },
   {
    "cell_type": "code",
    "execution_count": null,
@@ -45,6 +59,17 @@
     "condacolab.install()"
    ]
   },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "99fb1f76",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "# update conda\n",
+    "!conda update conda"
+   ]
+  },
   {
    "cell_type": "code",
    "execution_count": null,

diff --git a/torch_em/__version__.py b/torch_em/__version__.py
@@ -1 +1 @@
-__version__ = '0.1.0'
+__version__ = "0.4.0"