Skip to content

Instantly share code, notes, and snippets.

@jimmyahacker

jimmyahacker/create_venv.sh

Last active March 25, 2020 03:55
Show Gist options
  • Star 4 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save jimmyahacker/eeda0c25ae3068652e1ddf190be8744b to your computer and use it in GitHub Desktop.
Save jimmyahacker/eeda0c25ae3068652e1ddf190be8744b to your computer and use it in GitHub Desktop.
Download ZIP
Create virtual environment on NYU HPC Prince Cluster using tensorflow 1.7 or later in python3
Raw
create_venv.sh
# activate existed python3 module to get virtualenv
module load python3/intel/3.6.3
# create virtual environment with python3
virtualenv -p python3 $HOME/py3tf
# activate virtual environment
source $HOME/py3tf/bin/activate
# install tensorflow-gpu
pip install tensorflow-gpu
# unload module
module unload python3/intel/3.6.3
Raw
run_venv.sh
# load cuda and cudnn
module load cuda/9.0.176
module load cudnn/9.0v7.0.5
# activate virtual environment
source $HOME/py3tf/bin/activate
@jimmyahacker
Copy link
Author

jimmyahacker commented Apr 8, 2018 •
edited

Instructions

  1. Refer to this repo on how to login to NYU Prince Cluster
  2. After you login, first run source create_venv.sh and go out for a cup of coffee.
  3. Then you just run source run_venv.sh every time you wanna dive into this virtual environment and you could apply for CPU/GPU resources with this repo, too

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment