How to install caffe with python3 on Google Colab Notebook

Viewed 489

I am trying to install caffe on Google Colab.

!apt install caffe-cuda installs.

But in the next line when I type import caffe, it doesn't work. This error comes :


<ipython-input-8-a0183e542e82> in <module>()
      5 
      6 import sys
----> 7 import caffe

1 frames

/usr/lib/python3/dist-packages/caffe/pycaffe.py in <module>()
     11 import numpy as np
     12 
---> 13 from ._caffe import Net, SGDSolver, NesterovSolver, AdaGradSolver, \
     14         RMSPropSolver, AdaDeltaSolver, AdamSolver, NCCL, Timer
     15 import caffe.io

ModuleNotFoundError: No module named 'caffe._caffe'

Dont understand why this is happening.

0 Answers
Related