отсутствует библиотека в контейнере vitis-tool-gpu docker

Question

Я использовал контейнер инструментов Vitis docker, используя только процессор, и Conda работала нормально; однако, когда я хочу использовать версию GPU для docker, я получаю ошибку ниже. Я пытался создать среду дважды, и каждый раз ей не удавалось импортировать нужные библиотеки.

(vitis-ai-caffe) sam@Itec:~/cf_resnet50$ vai_q_caffe quantize -model float/trainval.prototxt -weights float/trainval.caffemodel
vai_q_caffe: error while loading shared libraries: libprotobuf.so.21: cannot open shared object file: No such file or director

Как я могу устранить эту ошибку? Я буду рад получить любую помощь.

Answer 1

В версии 1.0.0 существует проблема в docker с версией библиотеки protobuf. Это может быть решено путем исправления conda_requirements.txt. Для этого откройте conda_requirements.txt в текстовом редакторе и добавьте следующие две строки, затем соберите графический процессор docker:

libprotobuf==3.10.1
protobuf==3.10.1