Skip to content

Instantly share code, notes, and snippets.

@lesterlo
Forked from lukeyeager/install-digits-centos7.sh
Created February 8, 2017 16:49
Show Gist options
  • Save lesterlo/ab0f1d5c54c1b963176ec269d5235dec to your computer and use it in GitHub Desktop.
Save lesterlo/ab0f1d5c54c1b963176ec269d5235dec to your computer and use it in GitHub Desktop.
Install DIGITS and NVcaffe on CentOS 7
# nvidia-docker run -ti nvidia/cuda:7.0-devel-centos7
yum update
yum install epel-release
yum install gcc gcc-c++ git vim python-devel python-pip make cmake \
blas-devel lapack-devel openblas-devel \
protobuf-devel leveldb-devel snappy-devel opencv-devel boost-devel hdf5-devel gflags-devel glog-devel lmdb-devel libjpeg-devel libpng-devel
# Too old:
# yum install Cython scipy python-matplotlib
yum install numpy python-pandas python-imaging protobuf-python h5py
yum-builddep python-matplotlib
cd ~
git clone https://github.com/NVIDIA/caffe.git
cd caffe
cat python/requirements.txt | xargs -n1 pip install
mkdir build
cd build
cmake .. -DBLAS=Open
make -j12
# DIGITS
yum install python-gevent python-flask python-flask-wtf
cd ~
git clone https://github.com/NVIDIA/digits.git
cd digits
cat requirements.txt | xargs -n1 pip install
export CAFFE_HOME=~/caffe
./digits-devserver
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment