nogibjj / mlops-template

mlops template

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Migrate Tensorflow install into the Dockerfile vs in .devcontainer post install

noahgift opened this issue · comments

Currently Tensorflow for GPU is installed after the Docker image is built: https://github.com/nogibjj/mlops-template/blob/GPU/.devcontainer/devcontainer.json#L78, slowing things down. It needs to be moved into https://github.com/nogibjj/mlops-template/blob/GPU/.devcontainer/Dockerfile