COLAB中的CAFFE2

2024-04-29 00:28:24 发布

您现在位置:Python中文网/ 问答频道 /正文

我使用:!apt install -y caffe-cuda在colab中构建了caffe2。 当我导入caffe2时,它会正常导入,但当我导入caffe时,它会给出一个错误

当我做caffe.set_mode_gpu()时,它会给我这个错误:

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

Tags: installinpygpudemomode错误line