Difference between revisions of "Using Caffe On The Lab Computers"
Line 1: | Line 1: | ||
<pre> | <pre> | ||
− | Using a reference CaffeNet model from a student's Jupyter | + | Using a reference CaffeNet model from a student's Jupyter notebook code, we have found there are several things one needs to be aware of for the Caffe installation on our CS Lab computers: |
− | notebook code, we have found there are several things one needs to be aware of for the Caffe installation on our CS Lab computers: | ||
* You don't need to `import sys` and add things to `sys.path`. You can | * You don't need to `import sys` and add things to `sys.path`. You can |
Revision as of 15:20, 10 May 2016
Using a reference CaffeNet model from a student's Jupyter notebook code, we have found there are several things one needs to be aware of for the Caffe installation on our CS Lab computers: * You don't need to `import sys` and add things to `sys.path`. You can just `import caffe`. * You need to be running on one of the lab machines (ugrad1 through ugrad24; masters1 through masters16) in order to use GPU mode. Caffe will run on our compute systems (ugradx, gradx, gradz), but only in CPU mode; you won't be able to do the GPU section of the notebook on those three systems. * The systemwide installation *only* includes the Caffe binaries and python modules. It does not include the utility scripts and reference models. You will need to download those yourself. What you'll need to do is grab your own copy of the Caffe distribution in order to use their models: git clone https://github.com/BVLC/caffe.git Then, as the notebook indicates, run python from the examples directory: cd caffe/examples python The notebook's second input block should be as follows: caffe_root = '../' # this file should be run from {caffe_root}/examples (otherwise change this line) import caffe Everything else should work as expected.