python-2.7 - 尝试导入 caffe2.python 时出现分段错误

Question

尝试在 google colab 上使用 python = 2.7、cuda 10、cudnn 7.5 导入 caffe2.python 时出现分段错误

我尝试过使用 cuda 9，错误是：libcudart.9.so 无法打开共享对象

!wget https://anaconda.org/pytorch/pytorch/1.1.0/download/linux-64/pytorch-1.1.0-py2.7_cuda10.0.130_cudnn7.5.1_0.tar.bz2

!tar xvjf pytorch-1.1.0-py2.7_cuda10.0.130_cudnn7.5.1_0.tar.bz2

!cp -r lib/python2.7/site-packages/* /usr/local/lib/python2.7/dist-packages/

检查 Caffe2 构建是否成功

!python -c 'from caffe2.python import core' 2>/dev/null && echo "Success" || 回声“失败”

!python -c '从 caffe2.python 导入工作区；打印（工作区。NumCudaDevices（））'

错误是：

/bin/bash: line 1:
409 Segmentation fault (core dumped) python -c 'from caffe2.python import core' 2> /dev/null

失败

警告：root：此 caffe2 python 运行不支持 GPU。将在仅 CPU 模式下运行。

预期的 ：

成功

1