YoujiaZhang / AlphaGo-Zero-Gobang

Meta-Zeta是一个基于强化学习的五子棋(Gobang)模型,主要用以了解AlphaGo Zero的运行原理的Demo,即神经网络是如何指导MCTS做出决策的,以及如何自我对弈学习。源码+教程
MIT License
80 stars 8 forks source link

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

Open hrwangcn opened 1 year ago

hrwangcn commented 1 year ago

需要自行配置tensorflow config

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