机器学习
- | tiny-cnn | caffe | Theano | TensorFlow |
---|---|---|---|---|
Prerequisites | Nothing(Optional:TBB,OpenMP) | BLAS,Boost,protobuf,glog,gflags,hdf5, (Optional:CUDA,OpenCV,lmdb,leveldb etc) | Numpy,Scipy,BLAS,(optional:nose,Sphinx,CUDA etc) | numpy,six,protobuf,(optional:CUDA,Bazel) |
Modeling By | C++ code | Config File | Python Code | Python Code |
GPU Support | No | Yes | Yes | Yes |
Installing | Unnecessary | Necessary | Necessary | Necessary |
Windows Support | Yes | No* | Yes | No* |
Pre-Trained Model | Yes(via caffe-converter) | Yes | No* | No* |
-
*
unofficial version is available
FROM alpine:3.6
LABEL com.nvidia.volumes.needed="nvidia_driver"
ENV PATH /usr/local/nvidia/bin:/usr/local/cuda/bin:${PATH}
ENV LD_LIBRARY_PATH /usr/local/nvidia/lib:/usr/local/nvidia/lib64
RUN /bin/sh