New issue

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

#

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

Already on GitHub? # to your account

Jump to bottom

Attempting to perform BLAS operation using StreamExecutor without BLAS support #8

Open

hrwangcn opened this issue Nov 14, 2023 · 0 comments

hrwangcn commented Nov 14, 2023

需要自行配置tensorflow config

import tensorflow as tf
physical_devices = tf.config.list_physical_devices('GPU') 
tf.config.experimental.set_memory_growth(physical_devices[0], True)

# for free to join this conversation on GitHub. Already have an account? # to comment