Using Caffe in Python

113 views

Skip to first unread message

Mirza Ćutuk

unread,

Nov 28, 2022, 12:13:59 PM11/28/22

to Caffe Users

Hello everyone,

I used docker and this image https://hub.docker.com/r/ibmcom/caffe-ppc64le to set up caffe on an IBM machine. When I enter the container and write caffe -version, I get:

caffe version 1.0.0

which signals that caffe is good to go. However, when I import it in the python console and try to use any of its functions, I get the error that AttributeError: module 'caffe' has no attribute 'set_mode_cpu' (in the example of set_mode_cpu).