3 ms·
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?
by anonymousDan 5y ago
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 5y agoYou 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