Блог

Главная - Вопросы по программированию - Подготовка 3D разреженной свертки Op

Подготовка 3D разреженной свертки Op

#python #docker #tensorflow

Вопрос:

компиляция операций в контейнере docker, вот шаг, который я предпринял

 git clone https://github.com/tensorflow/tensorflow cd tensorflow amp;amp; git checkout v2.3.0 amp;amp; cd .. git clone https://github.com/tensorflow/custom-op --depth=1 export TF_FOLDER="/home/shegun/tensorflow" export CUSTOM_OP_FOLDER="/home/shegun/custom-op" mkdir -p tf3d/ops/third_party  cp -a ${TF_FOLDER}/third_party/eigen3 ${TF_FOLDER}/third_party/mkl  ${TF_FOLDER}/third_party/toolchains ${TF_FOLDER}/third_party/BUILD  ${TF_FOLDER}/third_party/eigen.BUILD  ${TF_FOLDER}/third_party/com_google_absl_fix_mac_and_nvcc_build.patch  ${TF_FOLDER}/third_party/com_google_absl.BUILD  ${TF_FOLDER}/third_party/cub.BUILD ${TF_FOLDER}/third_party/repo.bzl  tf3d/ops/third_party/  cp -a ${CUSTOM_OP_FOLDER}/gpu ${CUSTOM_OP_FOLDER}/tf  ${CUSTOM_OP_FOLDER}/configure.sh tf3d/ops/
  1. выполнил следующую команду docker pull tensorflow/tensorflow:2.3.0-custom-op-gpu-ubuntu16 docker run --rm --gpus -it -v /home/shegun/tf3d:/tf3d -w /tf3d/ops tensorflow/tensorflow:2.3.0-custom-op-gpu-ubuntu16
  2. В образе докера введите папку «tf3d/ops». Я выполнил следующее, чтобы проверить здание, чтобы проверить разреженность

    базель запускает sparse_conv_ops_py_test —экспериментальный_репо_ремоте_экзек —подробные сведения

  3. испытание прошло успешно. Я скопировал общую библиотеку в tf3d/ops/tensorflow_sparse_conv_ops папку

cp -a bazel-bin/tensorflow_sparse_conv_ops/_sparse_conv_ops.so tensorflow_sparse_conv_ops/ 5. Я выхожу из образа docker, вхожу в родительскую папку, содержащую tf3d папку, затем я попытался импортировать разреженные операции Conv следующим образом:

import tf3d.ops.tensorflow_sparse_conv_ops as sparse_conv_ops

вот ошибка, которую я получил : import-im6.q16: not authorized sparse_conv_ops’ @ error/constitute.c/WriteImage/1037.`

Не знаю, что делать. Мне нужна помощь. Спасибо

Метки: