只需要一行代码
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,即可解决问题
本文提供了一种在Ubuntu环境下快速安装Caffe-CUDA的方法,并解决了在Python3.7环境中导入Caffe模块时遇到的ModuleNotFoundError问题。通过修改文件名或回退至Python3.6,可以顺利运行Caffe。
5876

被折叠的 条评论
为什么被折叠?



