Skip to content

Marvin: A Minimalist GPU-only N-Dimensional ConvNets Framework

License

Notifications You must be signed in to change notification settings

fdoperezi/marvin

 
 

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

1 Commit
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

#Marvin#

Marvin is a GPU-only neural network framework made with simplicity, hackability, speed, memory consumption, and high dimensional data in mind.

#Dependences#

Download CUDA 7.5 and cuDNN 3. You will need to register with NVIDIA. Below are some additional steps to set up cuDNN 3:

CUDA_LIB_DIR=/usr/local/cuda/lib$([[ $(uname) == "Linux" ]] && echo 64)
echo LD_LIBRARY_PATH=$LD_LIBRARY_PATH:$CUDA_LIB_DIR >> ~/.profile && ~/.profile

tar zxvf cudnn*.tgz
sudo cp cuda/lib/* $CUDA_LIB_DIR
sudo cp cuda/include/* /usr/local/cuda/include

#Compilation#

./compile.sh

#MNIST#

  1. Prepare data: run examples/mnist/prepare_mnist.m in Matlab
  2. Train a model: run ./examples/mnist/demo.sh in shell
  3. Visualize filters: run examples/mnist/demo_vis_filter.m in Matlab

#Tutorials and Documentation# Please see our website at http://marvin.is.

About

Marvin: A Minimalist GPU-only N-Dimensional ConvNets Framework

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • C++ 98.6%
  • Cuda 1.3%
  • Shell 0.1%