Skip to content

Instantly share code, notes, and snippets.

@rpsene
Created April 21, 2019 20:22
Show Gist options
  • Save rpsene/7d9bdcf5999c53c14f8149c326321914 to your computer and use it in GitHub Desktop.
Save rpsene/7d9bdcf5999c53c14f8149c326321914 to your computer and use it in GitHub Desktop.
nvidia-docker run -it --env LICENSE=yes -p <PORT>:<PORT>/tcp ibmcom/powerai:1.6.0-all bash
python -c "import torch;print(torch.__version__)"
python -c "import tensorflow as tf;print(tf.__version__)"
conda install -y git jupyter matplotlib
git clone https://github.com/pvaneck/tf_mnist
cd tf_mnist
python ./train_basic_model.py
jupyter notebook --generate-config
jupyter notebook password
jupyter notebook --ip=0.0.0.0 --port=<PORT> --allow-root
@rpsene
Copy link
Author

rpsene commented Apr 22, 2019

python ./classify_mnist.py sample-images/img_1.jpg

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