跳到主要内容

机器学习

-tiny-cnncaffeTheanoTensorFlow
PrerequisitesNothing(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 ByC++ codeConfig FilePython CodePython Code
GPU SupportNoYesYesYes
InstallingUnnecessaryNecessaryNecessaryNecessary
Windows SupportYesNo*YesNo*
Pre-Trained ModelYes(via caffe-converter)YesNo*No*
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
docker build -t alpine-nvidia
nvidia-docker run -ti --rm alpine-nvidia

学习资源