Skip to content

Commit

Permalink
the pip install for TF2 needs updating
Browse files Browse the repository at this point in the history
TF2 installs GPU by default - need to drop the -gpu modifier to install correctly via pip in colab.

likely the same issue in all TF2 colab examples

PiperOrigin-RevId: 303945374
  • Loading branch information
uqtsherr authored and akhorlin committed Mar 31, 2020
1 parent b80b442 commit bae467d
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion examples/colab/tf2_arbitrary_image_stylization.ipynb
Original file line number Diff line number Diff line change
Expand Up @@ -90,7 +90,7 @@
},
"source": [
"# We want to use TensorFlow 2.0 in the Eager mode for this demonstration. But this module works as well with the Graph mode.\n",
"!pip install -U --pre tensorflow-gpu --quiet"
"!pip install tensorflow --quiet"
],
"execution_count": 0,
"outputs": []
Expand Down

0 comments on commit bae467d

Please sign in to comment.