Skip to content

Instantly share code, notes, and snippets.

@aminnj
Created July 1, 2019 22:19
Show Gist options
  • Save aminnj/9a9dda9827ca5da4ac9bbff9dffb9073 to your computer and use it in GitHub Desktop.
Save aminnj/9a9dda9827ca5da4ac9bbff9dffb9073 to your computer and use it in GitHub Desktop.
tensorflow with cuda 10.0 in docker w/ virtualenv
# on uaf, use singularity to load docker. with `--nv` for nvidia stuff
singularity shell --bind /cvmfs --nv docker://tensorflow/tensorflow:latest-gpu-py3
[ -d virtualenv ] || pip install --target=`pwd`/virtualenv virtualenv
[ -d myenv ] || virtualenv/bin/virtualenv -p `which python` myenv
source myenv/bin/activate
pip install pydicom pandas numpy matplotlib scikit-learn scikit-image opencv-python-headless tensorflow-gpu keras jupyter tqdm
export TF_FORCE_GPU_ALLOW_GROWTH=true
jupyter notebook --no-browser --port=8880
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment