mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-21 05:34:18 +08:00
30
README.md
30
README.md
@ -46,3 +46,33 @@ To run the tutorials you'll need ipython-notebooks and matplotlib, which can be
|
|||||||
|
|
||||||
brew install matplotlib --with-python3
|
brew install matplotlib --with-python3
|
||||||
pip install ipython notebook
|
pip install ipython notebook
|
||||||
|
|
||||||
|
## Build status (known working)
|
||||||
|
|
||||||
|
Ubuntu 14.04 (GCC)
|
||||||
|
- [ ] Default CPU build
|
||||||
|
- [x] Default GPU build
|
||||||
|
|
||||||
|
OS X (Clang)
|
||||||
|
- [x] Default CPU build
|
||||||
|
- [ ] Default GPU build
|
||||||
|
|
||||||
|
Options (both Clang and GCC)
|
||||||
|
- [ ] Nervana GPU
|
||||||
|
- [ ] ZMQ
|
||||||
|
- [ ] RocksDB
|
||||||
|
- [ ] MPI
|
||||||
|
- [ ] OpenMP
|
||||||
|
- [ ] No LMDB
|
||||||
|
- [ ] No LevelDB
|
||||||
|
- [x] No OpenCV
|
||||||
|
|
||||||
|
BLAS
|
||||||
|
- [x] OpenBLAS
|
||||||
|
- [x] ATLAS
|
||||||
|
- [ ] MKL
|
||||||
|
|
||||||
|
Other
|
||||||
|
- [ ] CMake 2.8 support
|
||||||
|
- [ ] List of dependencies for Ubuntu 14.04
|
||||||
|
- [ ] List of dependencies for OS X
|
||||||
|
@ -9,22 +9,28 @@ find_package(Threads REQUIRED)
|
|||||||
list(APPEND Caffe2_LINKER_LIBS ${CMAKE_THREAD_LIBS_INIT})
|
list(APPEND Caffe2_LINKER_LIBS ${CMAKE_THREAD_LIBS_INIT})
|
||||||
|
|
||||||
# ---[ BLAS
|
# ---[ BLAS
|
||||||
set(BLAS "Atlas" CACHE STRING "Selected BLAS library")
|
set(BLAS "Eigen" CACHE STRING "Selected BLAS library")
|
||||||
set_property(CACHE BLAS PROPERTY STRINGS "Atlas;OpenBLAS;MKL")
|
set_property(CACHE BLAS PROPERTY STRINGS "Atlas;OpenBLAS;MKL")
|
||||||
list(APPEND Caffe2_LINKER_LIBS cblas)
|
|
||||||
|
|
||||||
if(BLAS STREQUAL "Atlas")
|
if(BLAS STREQUAL "Eigen")
|
||||||
|
# Eigen is header-only and we do not have any dependent libraries
|
||||||
|
add_definitions(-DCAFFE2_USE_EIGEN_FOR_BLAS)
|
||||||
|
elseif(BLAS STREQUAL "Atlas")
|
||||||
find_package(Atlas REQUIRED)
|
find_package(Atlas REQUIRED)
|
||||||
include_directories(SYSTEM ${ATLAS_INCLUDE_DIRS})
|
include_directories(SYSTEM ${ATLAS_INCLUDE_DIRS})
|
||||||
list(APPEND Caffe2_LINKER_LIBS ${ATLAS_LIBRARIES})
|
list(APPEND Caffe2_LINKER_LIBS ${ATLAS_LIBRARIES})
|
||||||
|
list(APPEND Caffe2_LINKER_LIBS cblas)
|
||||||
elseif(BLAS STREQUAL "OpenBLAS")
|
elseif(BLAS STREQUAL "OpenBLAS")
|
||||||
find_package(OpenBLAS REQUIRED)
|
find_package(OpenBLAS REQUIRED)
|
||||||
include_directories(SYSTEM ${OpenBLAS_INCLUDE_DIR})
|
include_directories(SYSTEM ${OpenBLAS_INCLUDE_DIR})
|
||||||
list(APPEND Caffe2_LINKER_LIBS ${OpenBLAS_LIB})
|
list(APPEND Caffe2_LINKER_LIBS ${OpenBLAS_LIB})
|
||||||
|
list(APPEND Caffe2_LINKER_LIBS cblas)
|
||||||
elseif(BLAS STREQUAL "MKL")
|
elseif(BLAS STREQUAL "MKL")
|
||||||
find_package(MKL REQUIRED)
|
find_package(MKL REQUIRED)
|
||||||
include_directories(SYSTEM ${MKL_INCLUDE_DIR})
|
include_directories(SYSTEM ${MKL_INCLUDE_DIR})
|
||||||
list(APPEND Caffe2_LINKER_LIBS ${MKL_LIBRARIES})
|
list(APPEND Caffe2_LINKER_LIBS ${MKL_LIBRARIES})
|
||||||
|
list(APPEND Caffe2_LINKER_LIBS cblas)
|
||||||
|
add_definitions(-DCAFFE2_USE_MKL)
|
||||||
endif()
|
endif()
|
||||||
|
|
||||||
# ---[ Google-glog
|
# ---[ Google-glog
|
||||||
|
Reference in New Issue
Block a user