GPU not used. #652
Unanswered
Joeyabuki99
asked this question in
Q&A
Replies: 0 comments
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
-
Hi guys. I have a problem that is stopping me. In particular I have a classifier and I want to train it, but thes lines:
give me the following error:
2024-06-04 14:19:38.581524: I tensorflow/tsl/cuda/cudart_stub.cc:28] Could not find cuda drivers on your machine, GPU will not be used.
2024-06-04 14:19:38.636587: I tensorflow/tsl/cuda/cudart_stub.cc:28] Could not find cuda drivers on your machine, GPU will not be used.
2024-06-04 14:19:38.637139: I tensorflow/core/platform/cpu_feature_guard.cc:182] This TensorFlow binary is optimized to use available CPU instructions in performance-critical operations.
To enable the following instructions: AVX2 FMA, in other operations, rebuild TensorFlow with the appropriate compiler flags.
2024-06-04 14:19:39.893316: W tensorflow/compiler/tf2tensorrt/utils/py_utils.cc:38] TF-TRT Warning: Could not find TensorRT
if I run the nvidia-smi command is all ok and I can see the GPU, but for example if I do nvcc --verison I get 'command not found' . How can I resolve this problem?
Thank you!
Beta Was this translation helpful? Give feedback.
All reactions