在尝试通过pip安装tensorflow-gpu时遇到错误，提示DefaultAvgPoolingOp仅支持NCHW格式在CPU上。为了解决这个问题，建议使用conda来安装特定版本的tensorflow-gpu。如果conda安装过程中仍出现问题，可以尝试移除清华源后重新安装。确保正确配置GPU环境和依赖，以避免类似错误。
tensorflow.python.framework.errors_impl.InvalidArgumentError: Default AvgPoolingOp only supports NHWC on device type CPU的解决
不要用pip安装tensorflow-gpu，要使用conda install tensorflow-gpu==自己合适的版本
使用conda 安装 ，出现错误时，删除清华源通道，然后再次安装
扫一扫
评论 2
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
查看更多评论
添加红包
