forked from nitishsrivastava/deepnet
-
Notifications
You must be signed in to change notification settings - Fork 2
/
INSTALL.txt
60 lines (50 loc) · 2.37 KB
/
INSTALL.txt
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
#########################################
INSTALLATION INSTRUCTIONS FOR DEEPNET
Nitish Srivastava
University of Toronto.
#########################################
(1) DEPENDENCIES
- Numpy
- Scipy
- CUDA Toolkit and SDK.
Install the toolkit and SDK.
Set an environment variable CUDA_BIN to the path to the /bin directory of
the cuda installation and CUDA_LIB to the path to the /lib64 (or /lib)
directory. Also add them to PATH and LD_LIBARAY_PATH.
For example, add the following lines to your ~/.bashrc file
export CUDA_BIN=/usr/local/cuda-5.0/bin
export CUDA_LIB=/usr/local/cuda-5.0/lib64
export PATH=${CUDA_BIN}:$PATH
export LD_LIBRARY_PATH=${CUDA_LIB}:$LD_LIBRARY_PATH
- Protocol Buffers.
Available from http://code.google.com/p/protobuf/
Make sure that the PATH environment variable includes the directory that
contains the protocol buffer compiler - protoc.
For example,
export PATH=/usr/local/bin:$PATH
(2) COMPILING CUDAMAT AND CUDAMAT_CONV
DeepNet uses Vlad Mnih's cudamat library and Alex Krizhevsky's
cuda-convnet library. Some additional kernels have been
added. To compile the library -
- run make in the cudamat dir.
(3) SET ENVIRONMENT VARIABLES
- Add the path to cudamat to LD_LIBRARY_PATH. For example if
DeepNet is located in the home dir,
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:$HOME/deepnet/cudamat
- Add the path to DeepNet to PYTHONPATH. For example, if DeepNet is located in the
home dir,
export PYTHONPATH=$PYTHONPATH:$HOME/deepnet
(4) RUN EXAMPLES
- Download and extract the MNIST dataset from http://www.cs.toronto.edu/~nitish/deepnet/mnist.tar.gz
This dataset consists of labelled images of handwritten digits as numpy files.
- cd to the deepnet/deepnet/examples dir
- run
$ python setup_examples.py <path to mnist dataset> <output path>
This will modify the example models and trainers to use the specified paths.
- There are examples of different deep learning models. Go to any one and
execute runall.sh. For example, cd to deepnet/deepnet/examples/rbm and execute
$ ./runall.sh
This should start training an RBM model.
DeepNet has been tested on Ubuntu 12.04 using CUDA 4.2 and 5.0 on a variety of NVIDIA
GPUs (GTX-280, GTX-580, GTX-690, M2090, K-20x).
If you have trouble installing/running DeepNet, please create a relevant post on GitHub.