-
Notifications
You must be signed in to change notification settings - Fork 72
[MRG] caffe install on aws image #2
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Open
eickenberg
wants to merge
3
commits into
deeplearningparis:master
Choose a base branch
from
eickenberg:caffe
base: master
Could not load branches
Branch not found: {{ refName }}
Loading
Could not load tags
Nothing to show
Loading
Are you sure you want to change the base?
Some commits from the old base branch may be removed from the timeline,
and old review comments may become outdated.
Open
Changes from all commits
Commits
Show all changes
3 commits
Select commit
Hold shift + click to select a range
File filter
Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
There are no files selected for viewing
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,84 @@ | ||
## Refer to http://caffe.berkeleyvision.org/installation.html | ||
# Contributions simplifying and improving our build system are welcome! | ||
|
||
# cuDNN acceleration switch (uncomment to build with cuDNN). | ||
# USE_CUDNN := 1 | ||
|
||
# CPU-only switch (uncomment to build without GPU support). | ||
# CPU_ONLY := 1 | ||
|
||
# To customize your choice of compiler, uncomment and set the following. | ||
# N.B. the default for Linux is g++ and the default for OSX is clang++ | ||
# CUSTOM_CXX := g++ | ||
|
||
# CUDA directory contains bin/ and lib/ directories that we need. | ||
CUDA_DIR := /usr/local/cuda | ||
# On Ubuntu 14.04, if cuda tools are installed via | ||
# "sudo apt-get install nvidia-cuda-toolkit" then use this instead: | ||
# CUDA_DIR := /usr | ||
|
||
# CUDA architecture setting: going with all of them. | ||
# For CUDA < 6.0, comment the *_50 lines for compatibility. | ||
CUDA_ARCH := -gencode arch=compute_20,code=sm_20 \ | ||
-gencode arch=compute_20,code=sm_21 \ | ||
-gencode arch=compute_30,code=sm_30 \ | ||
-gencode arch=compute_35,code=sm_35 \ | ||
-gencode arch=compute_50,code=sm_50 \ | ||
-gencode arch=compute_50,code=compute_50 | ||
|
||
# BLAS choice: | ||
# atlas for ATLAS (default) | ||
# mkl for MKL | ||
# open for OpenBlas | ||
BLAS := open | ||
# Custom (MKL/ATLAS/OpenBLAS) include and lib directories. | ||
# Leave commented to accept the defaults for your choice of BLAS | ||
# (which should work)! | ||
BLAS_INCLUDE := /home/ubuntu/OpenBLAS | ||
BLAS_LIB := /home/ubuntu/OpenBLAS | ||
|
||
# This is required only if you will compile the matlab interface. | ||
# MATLAB directory should contain the mex binary in /bin. | ||
# MATLAB_DIR := /usr/local | ||
# MATLAB_DIR := /Applications/MATLAB_R2012b.app | ||
|
||
# NOTE: this is required only if you will compile the python interface. | ||
# We need to be able to find Python.h and numpy/arrayobject.h. | ||
# PYTHON_INCLUDE := /usr/include/python2.7 \ | ||
# /usr/lib/python2.7/dist-packages/numpy/core/include | ||
PYTHON_INCLUDE := /home/ubuntu/venv/include/python2.7/ \ | ||
/home/ubuntu/venv/local/lib/python2.7/site-packages/numpy/core/include/ | ||
# Anaconda Python distribution is quite popular. Include path: | ||
# Verify anaconda location, sometimes it's in root. | ||
# ANACONDA_HOME := $(HOME)/anaconda | ||
# PYTHON_INCLUDE := $(ANACONDA_HOME)/include \ | ||
# $(ANACONDA_HOME)/include/python2.7 \ | ||
# $(ANACONDA_HOME)/lib/python2.7/site-packages/numpy/core/include \ | ||
|
||
# We need to be able to find libpythonX.X.so or .dylib. | ||
PYTHON_LIB := /usr/lib | ||
# PYTHON_LIB := $(ANACONDA_HOME)/lib | ||
|
||
# Uncomment to support layers written in Python (will link against Python libs) | ||
# WITH_PYTHON_LAYER := 1 | ||
|
||
# Whatever else you find you need goes here. | ||
INCLUDE_DIRS := $(PYTHON_INCLUDE) /usr/local/include | ||
LIBRARY_DIRS := $(PYTHON_LIB) /usr/local/lib /usr/lib | ||
|
||
# Uncomment to use `pkg-config` to specify OpenCV library paths. | ||
# (Usually not necessary -- OpenCV libraries are normally installed in one of the above $LIBRARY_DIRS.) | ||
# USE_PKG_CONFIG := 1 | ||
|
||
BUILD_DIR := build | ||
DISTRIBUTE_DIR := distribute | ||
|
||
# Uncomment for debugging. Does not work on OSX due to https://github.com/BVLC/caffe/issues/171 | ||
# DEBUG := 1 | ||
|
||
# The ID of the GPU that 'make runtest' will use to run unit tests. | ||
TEST_GPUID := 0 | ||
|
||
# enable pretty build (comment to see full commands) | ||
Q ?= @ | ||
|
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,18 @@ | ||
#!/bin/bash | ||
|
||
sudo apt-get install protobuf-compiler libboost-all-dev libgflags-dev libgoogle-glog-dev libhdf5-serial-dev libleveldb-dev liblmdb-dev libsnappy-dev libopencv-dev libyaml-dev | ||
|
||
git clone https://github.com/BVLC/caffe.git ../caffe | ||
|
||
cp Makefile.config.example ../caffe/Makefile.config | ||
|
||
cd ../caffe | ||
make all | ||
make pycaffe | ||
|
||
cd python | ||
pip install networkx -U | ||
pip install pillow -U | ||
pip install -r requirements.txt | ||
|
||
ln -s /home/ubuntu/caffe/python/caffe /home/ubuntu/venv/lib/python2.7/site-packages/caffe |
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
(I made this script to be executed from the
dl-machine
folder. It would probably be more robust if I put absolute directories ... was my intention, but I forgot before making the PR)There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Have a look at similar instructions in
install-deeplearning-libraries.sh
.You can follow the pattern:
(cd caffee && run command)
to run the command inside the caffee folder and restore the CWD automatically to the previous state once done with that command.