CHTC / templates-GPUs

Template job submissions using GPUs in CHTC
MIT License
39 stars 11 forks source link

Add conda GPU example #8

Closed agitter closed 4 years ago

agitter commented 4 years ago

The adapts the existing PyTorch Docker example to run PyTorch inside a GPU-enabled conda environment.

agitter commented 4 years ago

Can we please merge this pull request if everything looks good? I may follow up with another pull request that will also use the shared PyTorch example files.