| commit | 0f870d4f409ea15242c4f87ff67c9ecdfdfb3dd3 | [log] [tgz] |
|---|---|---|
| author | Andrew Tulloch <tulloch@fb.com> | Wed Jan 25 17:36:39 2017 -0800 |
| committer | Facebook Github Bot <facebook-github-bot@users.noreply.github.com> | Wed Jan 25 17:44:22 2017 -0800 |
| tree | 2641ab913634ee8640e17e49685b72af17b5b1b4 | |
| parent | 9775ffc6ae468107a4bcc3e60ca1a3cd33e9f781 [diff] |
Add error checking for too-small input in ConvPoolOpBase Summary: Fixes segfaults that occur in Eigen and im2col/sgemm backends. Reviewed By: Yangqing Differential Revision: D4451772 fbshipit-source-id: 3cf21e5afb2fe300db4228933a82063db5f7091f
Caffe2 is a deep learning framework made with expression, speed, and modularity in mind. It is an experimental refactoring of Caffe, and allows a more flexible way to organize computation.
Caffe2 is released under the BSD 2-Clause license.
git clone --recursive https://github.com/bwasti/caffe2.git cd caffe2
brew install automake protobuf mkdir build && cd build cmake .. make
sudo apt-get install libprotobuf-dev protobuf-compiler libatlas-base-dev libgoogle-glog-dev libgtest-dev liblmdb-dev libleveldb-dev libsnappy-dev python-dev python-pip libiomp-dev libopencv-dev libpthread-stubs0-dev cmake
sudo pip install numpy
wget http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1404/x86_64/cuda-repo-ubuntu1404_8.0.44-1_amd64.deb
sudo dpkg -i cuda-repo-ubuntu1404_8.0.44-1_amd64.deb
sudo apt-get update
sudo apt-get install cuda
sudo apt-get install git
CUDNN_URL="http://developer.download.nvidia.com/compute/redist/cudnn/v5.1/cudnn-8.0-linux-x64-v5.1.tgz" &&
curl -fsSL ${CUDNN_URL} -O &&
sudo tar -xzf cudnn-8.0-linux-x64-v5.1.tgz -C /usr/local &&
rm cudnn-8.0-linux-x64-v5.1.tgz &&
sudo ldconfig
mkdir build && cd build
cmake ..
make
To run the tutorials you'll need ipython-notebooks and matplotlib, which can be installed on OS X with:
brew install matplotlib --with-python3 pip install ipython notebook
Ubuntu 14.04 (GCC)
OS X (Clang)
Options (both Clang and GCC)
BLAS
Other