mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-06 12:20:52 +01:00
Re-apply #266
Summary: Closes https://github.com/caffe2/caffe2/pull/404 Differential Revision: D4943280 Pulled By: Yangqing fbshipit-source-id: c0988598d8ccb8329feac88382686324b90d4d46
This commit is contained in:
parent
b905166362
commit
deb1327b6e
101
README.md
101
README.md
|
|
@ -1,18 +1,22 @@
|
|||
# Caffe2
|
||||
|
||||
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 a lightweight, modular, and scalable deep learning framework. Building on the original [Caffe](http://caffe.berkeleyvision.org), Caffe2 is designed with expression, speed, and modularity in mind.
|
||||
|
||||
## Questions and Feedback
|
||||
|
||||
Please use Github issues (https://github.com/caffe2/caffe2/issues) to ask questions, report bugs, and request new features.
|
||||
|
||||
Please participate in our survey (https://www.surveymonkey.com/r/caffe2). We will send you information about new releases and special developer events/webinars.
|
||||
|
||||
## License and Citation
|
||||
|
||||
Caffe2 is released under the [BSD 2-Clause license](https://github.com/Yangqing/caffe2/blob/master/LICENSE).
|
||||
|
||||
## Building Caffe2
|
||||
### Build Status
|
||||
|
||||
[](https://travis-ci.org/caffe2/caffe2)
|
||||
|
||||
[](https://ci.appveyor.com/project/Yangqing/caffe2)
|
||||
|
||||
|
||||
Detailed build matrix (hit refresh if you see icons not showing up due to heroku):
|
||||
|
||||
| Target | Status |
|
||||
|
|
@ -24,83 +28,14 @@ Detailed build matrix (hit refresh if you see icons not showing up due to heroku
|
|||
| Linux + MKL | [](https://travis-ci.org/caffe2/caffe2) |
|
||||
| Windows | [](https://ci.appveyor.com/project/Yangqing/caffe2) |
|
||||
|
||||
git clone --recursive https://github.com/caffe2/caffe2.git
|
||||
cd caffe2
|
||||
|
||||
#### OS X
|
||||
|
||||
brew install automake protobuf
|
||||
mkdir build && cd build
|
||||
cmake ..
|
||||
make
|
||||
### Further Resources on [Caffe2.ai](http://caffe2.ai)
|
||||
|
||||
#### Ubuntu
|
||||
|
||||
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 python-protobuf
|
||||
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
|
||||
|
||||
#### Android and iOS
|
||||
|
||||
We use CMake's Android and iOS ports to build native binaries that you can then integrate into your Android or Xcode projects. See scripts/build_android.sh and scripts/build_ios.sh for more details.
|
||||
|
||||
For Android, one can also use gradle to build Caffe2 directly with Android Studio. An example project can be found [here](https://github.com/bwasti/AICamera). Note that you may need to configure Android Studio so that it has the right SDK and NDK versions to build the code.
|
||||
|
||||
#### Raspberry Pi
|
||||
|
||||
For Raspbian, run scripts/build_raspbian.sh on the Raspberry Pi.
|
||||
|
||||
#### Tegra X1
|
||||
|
||||
To install Caffe2 on NVidia's Tegra X1 platform, simply install the latest system with the NVidia JetPack installer, and then run scripts/build_tegra_x1.sh on the Tegra device.
|
||||
|
||||
## Python support
|
||||
|
||||
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
|
||||
|
||||
## Build status (known working)
|
||||
|
||||
Ubuntu 14.04 (GCC)
|
||||
- [x] Default CPU build
|
||||
- [x] Default GPU build
|
||||
|
||||
OS X (Clang)
|
||||
- [x] Default CPU build
|
||||
- [x] Default GPU build
|
||||
|
||||
Options (both Clang and GCC)
|
||||
- [ ] Nervana GPU
|
||||
- [ ] ZMQ
|
||||
- [ ] RocksDB
|
||||
- [ ] MPI
|
||||
- [ ] OpenMP
|
||||
- [x] No LMDB
|
||||
- [x] No LevelDB
|
||||
- [x] No OpenCV
|
||||
|
||||
BLAS
|
||||
- [x] OpenBLAS
|
||||
- [x] ATLAS
|
||||
- [ ] MKL
|
||||
|
||||
Other
|
||||
- [x] CMake 2.8 support
|
||||
- [x] List of dependencies for Ubuntu 14.04
|
||||
- [x] List of dependencies for OS X
|
||||
* [Installation](http://caffe2.ai/docs/getting-started.html)
|
||||
* [Learn More](http://caffe2.ai/docs/learn-more.html)
|
||||
* [Upgrading to Caffe2](http://caffe2.ai/docs/caffe-migration.html)
|
||||
* [Datasets](http://caffe2.ai/docs/datasets.html)
|
||||
* [Model Zoo](http://caffe2.ai/docs/zoo.html)
|
||||
* [Tutorials](http://caffe2.ai/docs/tutorials.html)
|
||||
* [Operators Catalogue](http://caffe2.ai/docs/operators-catalogue.html)
|
||||
* [C++ API](http://caffe2.ai/doxygen-c/html/classes.html)
|
||||
* [Python API](http://caffe2.ai/doxygen-python/html/namespaces.html)
|
||||
|
|
|
|||
|
|
@ -29,7 +29,7 @@
|
|||
// as a track point for bugs, until we find a proper versioning cycle.
|
||||
|
||||
#define CAFFE2_VERSION_MAJOR 0
|
||||
#define CAFFE2_VERSION_MINOR 5
|
||||
#define CAFFE2_VERSION_MINOR 6
|
||||
#define CAFFE2_VERSION_PATCH 0
|
||||
#define CAFFE2_VERSION \
|
||||
(CAFFE2_VERSION_MAJOR * 10000 + CAFFE2_VERSION_MINOR * 100 + \
|
||||
|
|
|
|||
|
|
@ -1,3 +1,5 @@
|
|||
## @package pooling
|
||||
# Module caffe2.python.helpers.pooling
|
||||
## @package fc
|
||||
# Module caffe2.python.helpers.pooling
|
||||
from __future__ import absolute_import
|
||||
|
|
|
|||
|
|
@ -1,3 +1,5 @@
|
|||
## @package layer_test_util
|
||||
# Module caffe2.python.layer_test_util
|
||||
from __future__ import absolute_import
|
||||
from __future__ import division
|
||||
from __future__ import print_function
|
||||
|
|
|
|||
|
|
@ -1,3 +1,5 @@
|
|||
## @package optimizers
|
||||
# Module caffe2.python.layers.optimizers
|
||||
from __future__ import absolute_import
|
||||
from __future__ import division
|
||||
from __future__ import print_function
|
||||
|
|
|
|||
|
|
@ -1,3 +1,5 @@
|
|||
## @package uniform_sampling
|
||||
# Module caffe2.python.layers.uniform_sampling
|
||||
from __future__ import absolute_import
|
||||
from __future__ import division
|
||||
from __future__ import print_function
|
||||
|
|
|
|||
|
|
@ -1,3 +1,5 @@
|
|||
## @package predictor_exporter
|
||||
# Module caffe2.python.predictor.predictor_exporter
|
||||
from __future__ import absolute_import
|
||||
from __future__ import division
|
||||
from __future__ import print_function
|
||||
|
|
|
|||
|
|
@ -1,3 +1,5 @@
|
|||
## @package predictor_py_utils
|
||||
# Module caffe2.python.predictor.predictor_py_utils
|
||||
from __future__ import absolute_import
|
||||
from __future__ import division
|
||||
from __future__ import print_function
|
||||
|
|
|
|||
|
|
@ -1,3 +1,5 @@
|
|||
## @package serde
|
||||
# Module caffe2.python.predictor.serde
|
||||
from __future__ import absolute_import
|
||||
from __future__ import division
|
||||
from __future__ import print_function
|
||||
|
|
|
|||
|
|
@ -1,3 +1,5 @@
|
|||
## @package predictor_constants
|
||||
# Module caffe2.python.predictor_constants
|
||||
from __future__ import absolute_import
|
||||
from __future__ import division
|
||||
from __future__ import print_function
|
||||
|
|
|
|||
|
|
@ -1,3 +1,5 @@
|
|||
## @package diagnose_protobuf
|
||||
# Module scripts.diagnose_protobuf
|
||||
"""Diagnoses the current protobuf situation.
|
||||
|
||||
Protocol buffer needs to be properly installed for Caffe2 to work, and
|
||||
|
|
@ -11,6 +13,10 @@ Usage:
|
|||
python scripts/diagnose_protobuf.py
|
||||
"""
|
||||
|
||||
from __future__ import absolute_import
|
||||
from __future__ import division
|
||||
from __future__ import print_function
|
||||
from __future__ import unicode_literals
|
||||
import os
|
||||
import re
|
||||
from subprocess import Popen, PIPE
|
||||
|
|
|
|||
Loading…
Reference in New Issue
Block a user