Skip to content

Commit

Permalink
[USGS-R#146] add bit about gpu training in readme
Browse files Browse the repository at this point in the history
  • Loading branch information
jsadler2 committed Jan 21, 2022
1 parent 4efd791 commit 3fa06a2
Showing 1 changed file with 6 additions and 0 deletions.
6 changes: 6 additions & 0 deletions workflow_examples/readme.md
Original file line number Diff line number Diff line change
Expand Up @@ -21,6 +21,12 @@ Assuming you have `snakemake` installed, you would run the "basic" workflow with
snakemake -s Snakefile_basic.smk --configfile config.yml -j1
```

## Train a model on a GPU
To train a model using GPU, this line can be added to the `run` part of your training rule before the actual code that trains the model:
```python
os.system("module load analytics cuda10.1/toolkit/10.1.105"
```

## Links
Snakemake docs: [https://snakemake.readthedocs.io/en/stable/](https://snakemake.readthedocs.io/en/stable/)

Expand Down

0 comments on commit 3fa06a2

Please sign in to comment.