Perhaps the easiest way to import a Python library such as TensorFlow is to use PyPI. Simply go to https://pypi.org/ and search for TensorFlow. This will give you the information needed and the ability to look at different versions. The installation steps are as follows:
- Go to https://pypi.org/ and search for TensorFlow.
- Copy the name and version number you want in this format: tensorflow==1.14.0.
- In the Workspace tab of Databricks, right-click anywhere and from the dropdown, click on Create and then Library:
- On the Create Library page, select PyPI as the library source:
- Copy the name of the library and the version number and paste that into the Package section.
- Click Create.
If you have already created a cluster, you can attach TensorFlow to it. You can also have TensorFlow installed on all clusters.