init.sh 563 B

12345678910111213141516171819202122
  1. module load python/3.7.0
  2. python -m venv ~/base
  3. source ~/base/bin/activate
  4. pip install https://download.pytorch.org/whl/cu100/torch-1.0.0-cp37-cp37m-linux_x86_64.whl
  5. pip install torchvision
  6. # make a backup just in case
  7. cp ~/.bashrc ~/.bashrc.bak
  8. # update bashrc
  9. echo 'module load python/3.7.0' >> ~/.bashrc
  10. echo 'source ~/base/bin/activate' >> ~/.bashrc
  11. echo 'alias show-logs="watch tail -n 20 $(ls -rt | grep .out | tail -n 1)"' >> ~/.bashrc
  12. echo 'alias mdebug="msub -N skynet_1 -A jvb-000-ag -l nodes=1:gpus=1,walltime=01:00:00 -I -qtest"' >> ~/.bashrc