User Tools

Site Tools


cs501r_f2016:lab14

Differences

This shows you the differences between two versions of the page.

Link to this comparison view

Both sides previous revision Previous revision
cs501r_f2016:lab14 [2017/11/28 10:42]
wingated
cs501r_f2016:lab14 [2017/11/28 11:02] (current)
wingated
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>​
  
  
  
cs501r_f2016/lab14.txt ยท Last modified: 2017/11/28 11:02 by wingated