How to import caffe?

[Savitha2018]

I got this error

Traceback (most recent call last):
File "extract_RGB_feats.py", line 9, in
from cnn_util import *
File "cnn_util.py", line 4, in
import caffe
File "/content/drive/My Drive/S2VT/caffe/python/caffe/init.py", line 1, in
from .pycaffe import Net, SGDSolver, NesterovSolver, AdaGradSolver, RMSPropSolver, AdaDeltaSolver, AdamSolver, NCCL, Timer
File "/content/drive/My Drive/S2VT/caffe/python/caffe/pycaffe.py", line 13, in
from ._caffe import Net, SGDSolver, NesterovSolver, AdaGradSolver,
ImportError: No module named _caffe

[abhirup2691]

use this before running the file
!apt install caffe-cuda