Difference between revisions of "Using Caffe On The Lab Computers"
(More general information; formatting.) |
|||
| (One intermediate revision by one other user not shown) | |||
| Line 19: | Line 19: | ||
GPU acceleration (via CUDA) is supported on our lab machines (ugrad1 through ugrad24 on the ugraduate network; masters1 through masters16 on the grad/research network). | GPU acceleration (via CUDA) is supported on our lab machines (ugrad1 through ugrad24 on the ugraduate network; masters1 through masters16 on the grad/research network). | ||
| − | GPU acceleration is '''not''' supported on our compute systems (ugradx, gradx, and gradz); on those systems, you will only be able to use Caffe in CPU mode. | + | GPU acceleration is '''not''' supported on our compute systems (ugradx, ugradz, gradx, and gradz); on those systems, you will only be able to use Caffe in CPU mode. |
== Supported Python Versions == | == Supported Python Versions == | ||
| Line 25: | Line 25: | ||
Caffe will only work with Python 2.7. That's the default version on our systems, so if you're just running <code>python</code>, you should be fine. If you run <code>python3</code>, the Caffe module will not be available. (Caffe uses Google's protocol buffers to store its data and, as of May 2016, we unfortunately don't support the protobuf module for Python 3.) | Caffe will only work with Python 2.7. That's the default version on our systems, so if you're just running <code>python</code>, you should be fine. If you run <code>python3</code>, the Caffe module will not be available. (Caffe uses Google's protocol buffers to store its data and, as of May 2016, we unfortunately don't support the protobuf module for Python 3.) | ||
| − | [[Category:Software]] | + | [[Category:Software Available On The CS Linux Clients]] |
Latest revision as of 13:26, 13 April 2020
Caffe, the deep learning framework, is supported on the CS department's Linux clients, but there are a few things to be aware of.
Installation Location
Caffe is installed in /usr/local/caffe . That installation only includes the binaries and Python modules. Notably, it does not include the reference models or utility scripts.
If you want to use the Caffe-distributed models, you'll need to clone the Caffe repository yourself and use the contents of the models directory:
git clone https://github.com/BVLC/caffe.git
Python Module Location
The appropriate directory is already part of the system's Python path, so if you're using the Caffe Python module, you can just use import caffe without having to tell Python where to find that module.
GPU Support
GPU acceleration (via CUDA) is supported on our lab machines (ugrad1 through ugrad24 on the ugraduate network; masters1 through masters16 on the grad/research network).
GPU acceleration is not supported on our compute systems (ugradx, ugradz, gradx, and gradz); on those systems, you will only be able to use Caffe in CPU mode.
Supported Python Versions
Caffe will only work with Python 2.7. That's the default version on our systems, so if you're just running python, you should be fine. If you run python3, the Caffe module will not be available. (Caffe uses Google's protocol buffers to store its data and, as of May 2016, we unfortunately don't support the protobuf module for Python 3.)