ImageAI
ImageAI copied to clipboard
i created gpu for tensorflow but not working by gpu,,,using cpu
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/
no error on message side, no have solution?? or what i should??
Please give me advice.