search

ml-tooling / ml-workspace

🛠 All-in-one web-based IDE specialized for machine learning and data science.
https://mltooling.org/ml-workspace
Apache License 2.0
3.45k stars 452 forks source link

python package "tensorflow-gpu" should be "tensorflow" #33

Closed sim-san closed 3 years ago

sim-san commented 4 years ago

Describe the issue:

In the flavor "GPU" the python package "tensorflow-gpu" is installed here. This leads to problems if other python packages have the package "tensorflow" in their requirments.

Tensorflow changes the python package name from tensorflow-gpu to tensorflow. This package support CPU and GPU (https://www.tensorflow.org/install/pip)

Dockerfile: https://github.com/ml-tooling/ml-workspace/blob/ae85a0261322da44ec70ad15b1ad69adfd7e5f55/gpu-flavor/Dockerfile#L139

Technical details: