Ubuntu16.04 使用anaconda安装caffe2 报错解决

嘿芝麻

于 2018-10-19 08:21:00 发布

阅读量421

点赞数

CC 4.0 BY-SA版权

本文链接：https://blog.youkuaiyun.com/zw__chen/article/details/84576585

本文详细介绍了在不同CUDA和CuDNN版本下安装Caffe2的方法，包括如何通过conda安装pytorch-nightly以支持Caffe2。同时，针对安装过程中遇到的PackageNotFoundError，提供了更新conda以解决问题的步骤，并附带了测试安装成功的命令。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

在根据caffe2的官网，可以执行以下命令进行安装：

# For Caffe2 with CUDA 9 and CuDNN 7 support:
conda install pytorch-nightly -c pytorch

# For Caffe2 with CUDA 8 and CuDNN 7 support:
conda install pytorch-nightly cuda80 -c pytorch

笔者出现以下报错：

PackageNotFoundError: Package not found: '' Dependencies missing in current linux-64 channels: 
- pytorch-nightly -> mkl >=2018 
- pytorch-nightly -> python >=3.7,<3.8.0a0

解决办法：更新conda，执行命令conda update conda。

使用命令

cd ~ && python -c 'from caffe2.python import core' 2>/dev/null && echo "Success" || echo "Failure"

测试是否安装成功：

test