Skip to content
This repository has been archived by the owner on Jan 7, 2025. It is now read-only.

How Can I set Caffe CPU mode while installing the Ubuntu Pre-built package. #1182

Closed
vuhailuyen1991 opened this issue Oct 19, 2016 · 4 comments

Comments

@vuhailuyen1991
Copy link

Hi there, are there any possible ways to set CPU_ONLY option to one if I use the .deb package?

@lukeyeager
Copy link
Member

  1. There's not any way to use the GPU build of Caffe if you don't have a GPU. Caffe's GPU build is sloppy and queries CUDA even when it doesn't need to.
  2. But, you can still set DIGITS to CPU-only mode by setting CUDA_VISIBLE_DEVICES= (setting it to nothing).

@Bidski
Copy link

Bidski commented Oct 27, 2016

But, you can still set DIGITS to CPU-only mode by setting CUDA_VISIBLE_DEVICES= (setting it to nothing).

@lukeyeager is this documented somewhere? I just ages trying to find this and only found results relating to DIGITS-2

@lukeyeager
Copy link
Member

Even though we assume that most of our users do have a GPU, DIGITS still works if it doesn't detect any on your system.

Using CUDA_VISIBLE_DEVICES to trick your system into thinking there aren't any available GPUs is an unusual use case and not something we feel the need to document clearly. Besides, this envvar is a general CUDA feature - not specific to DIGITS.

P.S. Prior to #1091 there was another way to select GPUs using a configuration file. But we chose to just follow standard practice now and use the CUDA_VISIBLE_DEVICES envvar.

@zahidmadeel
Copy link

Hello,
I have just installed digits from this link. I have downloaded the MNIST dataset as explained in getting started section but when I go on to build the model, it gives me error as asked in this question. I have already built caffe with CPU_ONLY Flag set o 1 in MakeFile.config. Why is digits then showing me the error. I have no nvidia GPU so I did not install any drivers or cuda. Is it required at all? How can I get around this issue?
Regards
Adeel

Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.
Projects
None yet
Development

No branches or pull requests

4 participants