saymrwulf/pytorch
1
0
Fork 0
mirror of https://github.com/saymrwulf/pytorch.git synced 2026-05-14 20:57:59 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions
Tensors and Dynamic neural networks in Python with strong GPU acceleration
658 commits 5021 branches 1206 tags 1 GiB
Python 56.8%
C++ 35%
Cuda 3%
C 1.5%
Objective-C++ 1.1%
Other 2.3%
Find a file
Simon Layton b99ea43c9a Set default build type to release 
Summary: Closes https://github.com/caffe2/caffe2/pull/89

Reviewed By: bwasti

Differential Revision: D4392954

Pulled By: Yangqing

fbshipit-source-id: 00ec72838e5e7dd9ff96449a8589273c68d0cef5
2017-01-09 10:59:39 -08:00
.travis mac travis: use Eigen instead of openblas 2017-01-04 22:17:18 -08:00
caffe/proto goodbye old brewery 2017-01-04 20:58:35 -08:00
caffe2 Linter fix up to sync fbsource and github 2017-01-06 15:36:17 -08:00
cmake fix typo 2017-01-05 10:48:26 -08:00
docs Update docs to reflect current build status 2017-01-05 09:59:31 -08:00
third_party goodbye old brewery 2017-01-04 20:58:35 -08:00
.Doxyfile Adding optional Eigen code. Added a switch USE_SYSTEM_EIGEN in Env. Misc changes. 2015-10-18 16:55:24 -07:00
.gitignore Reset and warn when any options are not satisfied 2017-01-04 15:46:43 -08:00
.gitmodules add back third_party/protobuf, but it won't be used in normal builds. 2017-01-04 17:27:18 -08:00
.travis.yml Update travis to test more versions of GCC and fix README build status link 2017-01-08 21:29:35 -08:00
CMakeLists.txt Set default build type to release 2017-01-09 10:59:39 -08:00
LICENSE adding license back 2017-01-05 10:08:17 -08:00
Makefile goodbye old brewery 2017-01-04 20:58:35 -08:00
README.md Update travis to test more versions of GCC and fix README build status link 2017-01-08 21:29:35 -08:00

README.md

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.

License and Citation

Caffe2 is released under the BSD 2-Clause license.

Building Caffe2

Build Status

git clone --recursive https://github.com/bwasti/caffe2.git
cd caffe2

OS X

brew install gtest automake protobuf
mkdir build && cd build
cmake ..
make

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
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

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)

  • Default CPU build
  • Default GPU build

OS X (Clang)

  • Default CPU build
  • Default GPU build

Options (both Clang and GCC)

  • Nervana GPU
  • ZMQ
  • RocksDB
  • MPI
  • OpenMP
  • No LMDB
  • No LevelDB
  • No OpenCV

BLAS

  • OpenBLAS
  • ATLAS
  • MKL

Other

  • CMake 2.8 support
  • List of dependencies for Ubuntu 14.04
  • List of dependencies for OS X