只需要一行代码
sudo apt install caffe-cuda
若import找不到caffe
import sys
sys.path.append('/usr/lib/python3/dist-packages')
import caffe
若使用python 3.7,会有错误提示 ModuleNotFoundError: No module named 'caffe._caffe',可安装并使用python 3.6版本
或把 /usr/lib/python3/dist-packages/caffe 下的 _caffe.cpython-36m-x86_64-linux-gnu.so 更名为 _caffe.cpython-37m-x86_64-linux-gnu.so,即可解决问题