gpu环境安装参考https://blog.csdn.net/qq_29153321/article/details/87203888
单机tensorflow cpu/gpu切换
python3 model.py(默认gpu下运行)
CUDA_VISIBLE_DEVICES="cpu" python3 model.py(切换到gpu下运行)
gpu环境安装参考https://blog.csdn.net/qq_29153321/article/details/87203888
单机tensorflow cpu/gpu切换
python3 model.py(默认gpu下运行)
CUDA_VISIBLE_DEVICES="cpu" python3 model.py(切换到gpu下运行)