无法在 google colaboratory 上运行 caffe2

Question

在完成!apt install caffe-cuda 之后，我尝试在 google colaboratory 上使用 caffe，但没有结果。欢迎任何帮助。 谢谢

Answer 1

apt install caffe-cuda 命令为 Python3 安装 cuda。 对于 python2，您必须手动安装它（克隆 git folder 并制作）。 在 colab 笔记本中使用以下命令：

#Clone the repo
!cp caffe/Makefile.config.example caffe/Makefile.config

# Make changes to Makefile
!git clone https://github.com/BVLC/caffe.git
!sed -i 's/# OPENCV_VERSION/OPENCV_VERSION/g' caffe/Makefile.config
!sed -i 's/CUDA_ARCH :=/CUDA_ARCH :=\n/g' caffe/Makefile.config
!sed -i 's/# USE_CUDNN/USE_CUDNN/g' caffe/Makefile.config
!sed -i 's/# WITH_PYTHON_LAYER := 1/WITH_PYTHON_LAYER := 1\n/g' caffe/Makefile.config
!sed -i 's/-gencode arch=compute_20,code=sm_20/#-gencode arch=compute_20,code=sm_20/g' caffe/Makefile.config
!sed -i 's/-gencode arch=compute_20,code=sm_21/#-gencode arch=compute_20,code=sm_21/g' caffe/Makefile.config
!sed -i 's/\/usr\/local\/include/\/usr\/local\/include \/usr\/include\/hdf5\/serial/g' caffe/Makefile.config
!sed -i 's/\/usr\/local\/lib/\/usr\/local\/lib \/usr\/lib\/x86_64-linux-gnu\/hdf5\/serial/g' caffe/Makefile.config
!sed -i 's/\/usr\/lib\/python2.7\/dist-packages\/numpy/\/usr\/local\/lib\/python2.7\/dist-packages\/numpy/g' caffe/Makefile.config
!cat caffe/Makefile.config

#Install Dependencies
!sudo apt-get install libprotobuf-dev libleveldb-dev libsnappy-dev libopencv-dev libhdf5-serial-dev protobuf-compiler libgflags-dev libgoogle-glog-dev liblmdb-dev

#Make
!cd caffe && make -j32

另外，我正在尝试安装 pycaffe。所以一旦你制作了pycaffe，你必须将caffe的python/文件夹添加到sys。您可以使用：

import sys
sys.path.insert(1, "{path to pycaffe}/python")