最新tensorflow用不了GPU,cuda对应的tensorflow版本
我的电脑已经安装了cuda和cudnn,也安装了tensorflow-gpu,但是跑的时候无法调用GPU运算 通过一下代码检测GPU是否被使用:import osfrom tensorflow.python.client import device_libos.environ["TF_CPP_MIN_LOG_LEVEL"] = "99"if __name__ == "__main__": print(device_lib.list_local_devices())output[na.
原创
2021-10-09 21:32:29 ·
4510 阅读 ·
10 评论