This shows you the differences between two versions of the page.
| Both sides previous revision Previous revision Next revision | Previous revision | ||
|
cs501r_f2016:lab14 [2017/11/28 17:42] wingated |
cs501r_f2016:lab14 [2021/06/30 23:42] (current) |
||
|---|---|---|---|
| Line 173: | Line 173: | ||
| </code> | </code> | ||
| + | ---- | ||
| + | ====Pytorch on the supercomputer:==== | ||
| + | |||
| + | The folks at the supercomputer center have installed `pytorch` and `torchvision`. | ||
| + | To use pytorch, you'll need to use the following modules in your SLURM file: | ||
| + | |||
| + | <code bash> | ||
| + | # these are dependencies: | ||
| + | module load cuda/8.0 cudnn/6.0_8.0 | ||
| + | module load python/27 | ||
| + | |||
| + | module load python-pytorch python-torchvision | ||
| + | |||
| + | </code> | ||
| + | |||
| + | If you need more python libraries you can install them to your home directory with: | ||
| + | |||
| + | <code bash> | ||
| + | pip install --user libraryname | ||
| + | </code> | ||