在根據(jù)caffe2的官網(wǎng)，可以執(zhí)行以下命令進(jìn)行安裝：

# 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

筆者出現(xiàn)以下報(bào)錯(cuò)：

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

解決辦法：更新conda，執(zhí)行命令conda update conda。

使用命令

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

測試是否安裝成功：

test