Question

Ошибка сегментации при попытке импортировать caffe2.python в google colab с python = 2.7, cuda 10, cudnn 7.5

Я пробовал с cuda 9, и ошибка: libcudart.9.so не может открыть общий объект

! 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 'изcaffe2.python import core '2> / dev / null && echo "Success" ||echo "Failure"

! python -c 'из рабочей области импорта caffe2.python;print (workspace.NumCudaDevices ()) '

Ошибка:

/ bin / bash: строка 1:
409 Ошибка сегментации (дамп ядра) python -c' из caffe2.pythonimport core '2> / dev / null

Failure

ПРЕДУПРЕЖДЕНИЕ: root: Этот запуск python caffe2 не поддерживает GPU.Будет работать только в режиме процессора.

Ожидается:

Успех

1

Ошибка сегментации при попытке импортировать caffe2.python