我可以在Google合作实验室上安装caffe或caffe2吗？

Question

我有尝试!pip install caffe和!pip install caffe2，但它没有用。

如果可以，如何安装caffe或caffe2？那其他框架呢？谢谢。

Answer 1

要安装具有GPU支持的Caffe，请尝试运行：

!apt install -y caffe-cuda

对于仅CPU版本：

!apt install -y caffe-cpu