ImageAI icon indicating copy to clipboard operation
ImageAI copied to clipboard

i created gpu for tensorflow but not working by gpu,,,using cpu

Open Fhwang0926 opened this issue 3 years ago • 0 comments

import tensorflow as tf
config = tf.compat.v1.ConfigProto(gpu_options = tf.compat.v1.GPUOptions(per_process_gpu_memory_fraction=0.8)
#device_count = {'GPU': }
)
config.gpu_options.allow_growth = True
session = tf.compat.v1.Session(config=config)
tf.compat.v1.keras.backend.set_session(session)
# Fix CUBLAS_STATUS_NOT_INITIALIZED error.
#  - failed to create cublas handle: CUBLAS_STATUS_NOT_INITIALIZED
os.environ["TF_FORCE_GPU_ALLOW_GROWTH"] = "true"

using option and set more env is

export PATH=$PATH:/usr/local/cuda-11.5/bin \
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-11.5/lib64 \
export CUDADIR=/usr/local/cuda-11.5/

what_happen

no error on message side, no have solution?? or what i should??

Please give me advice.

Fhwang0926 avatar Oct 24 '21 14:10 Fhwang0926