search

neuro-inc / neuro-base-environment

Base docker image used in Neuro Platform Template, deployed on DockerHub as neuromation/base
Apache License 2.0
1 stars 0 forks source link

Base image improvements: feedback by atselousov #93

Closed atemate closed 3 years ago

atemate commented 4 years ago 
I would update PyTorch version to 1.3 / 1.4.
Do we really need a separate TensorboardX package installation? As I know, it's part of PyTorch now (torch.utils.tensorboard).
Maybe it has the sense to delete wandb from the base image and make the installation by a flag in cookiecutter? Also, it can simplify Makefile in this case. It is really annoying to delete wandb files from every inited template if you don't use it.
Do we really need to build CMake?

By @atselousov . origin: https://neuromation.slack.com/archives/CGP770K27/p1582273220132600?thread_ts=1582107299.115400&cid=CGP770K27

YevheniiSemendiak commented 3 years ago

pythorch updated to 1.9 TensorboardX - easier to bump versions explicitly, not relying on pytorch wandb removing in #147 we don't build cmake anymore