AutoMachineLearning的起步

首先使用autokeras入个门
autokeras的依赖是tensorflow和pytorch,由于我已经安装了cuda和cudann,所以pytorch我是自行上官网安装的,至于tensorflow有个比较大的问题,我原本是已经安装了tensorflow2.0.0-gpu,但是它的依赖是cpu版本的…在我选择了pip安装autokeras之后(通过源码安装失败了,tf报错),就会出现
Your CPU supports instructions that this TensorFlow binary was not compiled to use: AVX AVX2
原因是下载TensorFlow的版本不支持cpu的AVX2编译,原因是默认下载X86_64的SIMD版本。换句话说,我使用gpu加速的能力丧失了…
非常窒息,等我找到更好的工具就把版本给滚回去,下面的做法就是可以屏蔽这个消息

import os  
os.environ["TF_CPP_MIN_LOG_LEVEL"]='1' # 这是默认的显示等级,显示所有信息  
os.environ["TF_CPP_MIN_LOG_LEVEL"]='2' # 只显示 warning 和 Error   
os.environ["TF_CPP_MIN_LOG_LEVEL"]='3' # 只显示 Error

或者说可以去官网安装在这里插入图片描述
我有时间可以考虑尝试一波,但我现在只想赶紧跑起来看看效果就得了。我想把我毕业设计的重点放在真正的算法设计上,而不是一个工具折腾来折腾去,没必要。

你可能感兴趣的:(随笔流水账)