单机tensorflow cpu/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下运行)

发布了29 篇原创文章 · 获赞 5 · 访问量 1万+

猜你喜欢

转载自blog.csdn.net/qq_29153321/article/details/88979895