![]() For example, as of April 23, 2020, TensorFlow's CUDA setting is theįollowing: # This config refers to building with CUDA available. To make MediaPipe get TensorFlow's CUDA settings, find TensorFlow'sĬopy the build:using_cuda and build:cuda section into MediaPipe's. usr/include into $TF_CUDA_PATHS for cudablas and libcudnn. That the following code snippet also adds /usr/lib/x86_64-linux-gnu and Setting $TF_CUDA_PATHS is the way to declare where the CUDA library is. $ ls /usr/lib/x86_64-linux-gnu/ | grep libcudnn.so README doc include libnsight nsightee_plugins nvvm share targets version.txtĬopyright (c) 2005-2019 NVIDIA CorporationĬuda compilation tools, release 10.1, V10.1.243 LICENSE bin extras lib64 libnvvp nvml samples src tools Libcupti.so.10.1 libcupti_static.a libnvperf_host_static.a Libcupti.so libcupti.so.10.1.208 libnvperf_host.so libnvperf_target.so It's recommended to verify the installation of CUPTI, CUDA, CuDNN, and NVCC: $ ls /usr/local/cuda/extras/CUPTI With: $ export PATH=/usr/local/cuda-10.1/bin$ To install the required NVIDIA software on your Linux desktop.Īfter installation, update $PATH and $LD_LIBRARY_PATH and run ldconfig To enable TensorFlow GPU inference with MediaPipe, the first step is to follow MediaPipe can work with TensorFlow to perform GPU inference on video cards that MediaPipe framework doesn't require CUDA for GPU compute and rendering. TensorFlow CUDA Support and Setup on Linux Desktop Systems (Apple doesn't support OpenGL ES 3.1+). Note: MEDIAPIPE_DISABLE_GL_COMPUTE is already defined automatically on all Apple If only ES 3.0 or below is supported, you can still build MediaPipe targets thatĭon't require TFLite inference on GPU with: $ bazel build -copt -DMESA_EGL_NO_X11_HEADERS -copt -DEGL_NO_X11 -copt -DMEDIAPIPE_DISABLE_GL_COMPUTE With this setup, build with: $ bazel build -copt -DMESA_EGL_NO_X11_HEADERS -copt -DEGL_NO_X11 You need to see ES 3.1 or greater printed in order to perform TFLite inference Try re-establishing your SSH connection with the -X option and try again. GPU information you see the output: glxinfo | grep -i opengl If you have connected to your computer through SSH and find when you probe for OpenGL ES profile shading language version string: OpenGL ES GLSL ES 3.20 OpenGL ES profile version string: OpenGL ES 3.2 NVIDIA 430.50 To check if your Linux desktop GPU can run MediaPipe with OpenGL ES: $ sudo apt-get install mesa-common-dev libegl1-mesa-dev libgles2-mesa-devįor example, it may print: $ glxinfo | grep -i opengl GPU compute and rendering and perform TFLite inference on GPU. On Linux desktop with video cards that support OpenGL ES 3.1+, MediaPipe can run Note: On Android and iOS, OpenGL ES is required by MediaPipe framework and the Should disable OpenGL ES support with: $ bazel build -define MEDIAPIPE_DISABLE_GPU=1 On platforms where OpenGL ES is not available (see also Disable OpenGL ES Supportīy default, building MediaPipe (with no special bazel flags) attempts to compileĪnd link against OpenGL ES (and for iOS also Metal) libraries. Machine learning inference calculators and graphs. OpenGL ES 3.1 or greater is required (on Android/Linux systems) for running In addition, MediaPipe also supports Metal on iOS. MediaPipe supports OpenGL ES up to version 3.2 on Android/Linux and up to ES 3.0
0 Comments
Leave a Reply. |