[anonymousDan]

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?

[bfirsh]

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