Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

GPU was not used for live demo #114

Open
Ishihara-Masabumi opened this issue Oct 20, 2023 · 1 comment
Open

GPU was not used for live demo #114

Ishihara-Masabumi opened this issue Oct 20, 2023 · 1 comment
Assignees
Labels
nvidiaproblems NVIDIA/CUDA problems with host system tensorflow problems Problems caused by the tensorflow framework

Comments

@Ishihara-Masabumi
Copy link

I tried to run some live demo, but I found that the execution was very slow and GPU was not used.
My environment is as follows:

Ubuntu20.04
Python3.8
tensorflow2.12.0
NVIDIA-SMI 530.30.02  Driver Version: 530.30.02    CUDA Version: 12.1

Please check this environment is OK or not, for live demo with GPU.

@AmmarkoV AmmarkoV self-assigned this Dec 7, 2023
@AmmarkoV AmmarkoV added nvidiaproblems NVIDIA/CUDA problems with host system tensorflow problems Problems caused by the tensorflow framework labels Dec 7, 2023
@AmmarkoV
Copy link
Collaborator

AmmarkoV commented Dec 7, 2023

I think that the CUDA version is very new and the libtensorflow provided does not properly utilize it

Taking a look on the tables here :
https://www.tensorflow.org/install/source

For CUDA 12.2 you want tensorflow-2.15.0 Built with Clang 16.0.0, Bazel 6.1.0 and CUDNN 8.8

Try downloading
https://storage.googleapis.com/tensorflow/libtensorflow/libtensorflow-gpu-linux-x86_64-2.15.0.tar.gz

and extracting it in the dependencies/ directory!

You can also change
from
TENSORFLOW_VERSION="2.3.1"
to
TENSORFLOW_VERSION="2.15.0"
in
https://github.com/FORTH-ModelBasedTracker/MocapNET/blob/master/initialize.sh#L209
and then run the initialize.sh script selecting Y when asked for GPU

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
nvidiaproblems NVIDIA/CUDA problems with host system tensorflow problems Problems caused by the tensorflow framework
Projects
None yet
Development

No branches or pull requests

2 participants