Upgrade Cuda drivers in simple audio tutorial.

markmcd · copybara-github · commit 993919c737bd · 2022-09-19T18:38:01.000-07:00
This is required to support TF 2.10 in Colab while it currently ships with a lower version. Fixes [googlecolab/colabtools#3058](googlecolab/colabtools#3058). PiperOrigin-RevId: 475435531
diff --git a/site/en/tutorials/audio/simple_audio.ipynb b/site/en/tutorials/audio/simple_audio.ipynb
@@ -98,8 +98,9 @@
       },
       "outputs": [],
       "source": [
-        "# Upgrading Colab's tensorflow breaks GPU support. Switch this notebook back to GPU once Colab upgrades.\n",
-        "!pip install -U --pre tensorflow tensorflow_datasets"
+	"# Upgrade environment to support TF 2.10 in Colab\n",
+        "!pip install -U --pre tensorflow tensorflow_datasets\n",
+	"!apt install --allow-change-held-packages libcudnn8=8.1.0.77-1+cuda11.2"
       ]
     },
     {

Original file line number	Diff line number	Diff line change
`@@ -98,8 +98,9 @@`
`98`	`98`	`},`
`99`	`99`	`"outputs": [],`
`100`	`100`	`"source": [`
`101`		`- "# Upgrading Colab's tensorflow breaks GPU support. Switch this notebook back to GPU once Colab upgrades.\n",`
`102`		`- "!pip install -U --pre tensorflow tensorflow_datasets"`
	`101`	`+ "# Upgrade environment to support TF 2.10 in Colab\n",`
	`102`	`+ "!pip install -U --pre tensorflow tensorflow_datasets\n",`
	`103`	`+ "!apt install --allow-change-held-packages libcudnn8=8.1.0.77-1+cuda11.2"`
`103`	`104`	`]`
`104`	`105`	`},`
`105`	`106`	`{`