Cog: Containers for Machine Learning

Re:Cog: Containers for Machine Learning by anonymousDan on Thursday April 21, 2022 @07:54AM (#31108296)

Is there an easy way to create a similar dev environment for training on Linux that will take care of all the CUDA driver nonsense?

[ Parent | on HN ]

Re:Cog: Containers for Machine Learning by bfirsh on Thursday April 21, 2022 @09:44AM (#31109281)

You can do this with Cog! Once you've written cog.yaml, you can run arbitrary commands inside the environment which has CUDA installed correctly:

  $ cog run python train.py

[ Parent | on HN ]