时间戳

首页 > 解决方案 > COLAB 中的 CAFFE2

python - COLAB 中的 CAFFE2

问题描述

我已经使用 colab 在 colab 中构建了 caffe2 !apt install -y caffe-cuda：当我导入 caffe2 时，它会正常导入，但是当我导入 caffe 时，它​​会出错。

当我这样做时caffe.set_mode_gpu()，它给了我这个错误：

File "demo.py", line 132, in <module>
    caffe.set_mode_gpu()
NameError: name 'caffe' is not defined

标签： python

解决方案

推荐阅读