Caffe2 - Python API
A deep learning, cross platform ML framework
Public Member Functions | Public Attributes | Static Public Attributes | List of all members
setup_caffe2.cmake_build Class Reference
Inheritance diagram for setup_caffe2.cmake_build:
setup_caffe2.Caffe2Command

Public Member Functions

def initialize_options (self)
 
def finalize_options (self)
 
def run (self)
 
- Public Member Functions inherited from setup_caffe2.Caffe2Command
def initialize_options (self)
 
def finalize_options (self)
 

Public Attributes

 jobs
 

Static Public Attributes

list user_options
 
bool built = False
 
- Static Public Attributes inherited from setup_caffe2.Caffe2Command
list user_options = []
 

Detailed Description

Compiles everything when `python setup.py build` is run using cmake.

Custom args can be passed to cmake by specifying the `CMAKE_ARGS`
environment variable. E.g. to build without cuda support run:
    `CMAKE_ARGS=-DUSE_CUDA=Off python setup.py build`

The number of CPUs used by `make`/`ninja` can be specified by passing
`-j<ncpus>` to `setup.py build`.  By default all CPUs are used.

Definition at line 98 of file setup_caffe2.py.

Member Data Documentation

list setup_caffe2.cmake_build.user_options
static
Initial value:
1 = [
2  (str('jobs='), str('j'),
3  str('Specifies the number of jobs to use with make or ninja'))
4  ]

Definition at line 109 of file setup_caffe2.py.


The documentation for this class was generated from the following file: