Navigation Menu

Skip to content

Instantly share code, notes, and snippets.

@shnhrtkyk

shnhrtkyk/job_pytorch.sh

Created January 17, 2020 23:03
Show Gist options
  • Star 2 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save shnhrtkyk/7c5ad37b7fcfa36888117c7ed53cefad to your computer and use it in GitHub Desktop.
Save shnhrtkyk/7c5ad37b7fcfa36888117c7ed53cefad to your computer and use it in GitHub Desktop.
Download ZIP
TSUBAMEでpytorch
Raw
job_pytorch.sh
#!/bin/sh
#$ -cwd
#$ -l q_node=1
#$ -l h_rt=24:00:00
. /etc/profile.d/modules.sh
module load python/3.6.5
module load cuda/10.1.105
module load nccl/2.2.13
module load intel/19.0.0.117
module load cudnn/7.4
pip install --user --upgrade pip
pip3 install --user laspy
pip3 install --user scikit-learn
pip3 install --user matplotlib
pip3 install --user torch torchvision
python hoge.py
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment