autoKeras
autoKeras GitHub:https://github.com/jhfjhfj1/autokeras
百度網盤下載地址:http://pandownload.com/
autoKeras官方文檔:https://autokeras.com/
開始進入主題:Windows安裝autoKeras
第一步:安裝autoKeras:https://github.com/jhfjhfj1/autokeras 這里是壓縮包,使用 python setup.py install
第二步:依賴於pytorch 所以需要安裝 torch
版本對比:torch0.4-cuda9.0-python3.6
torch 見博客。此處需要翻牆https://pytorch.org/,
cuda9 博客
第三步:pip3 install torchvision
過程中遇到的錯誤或許能幫助你,更新numpy到新 numpy+mke 保存可視化圖片需安裝:https://www.cnblogs.com/shuodehaoa/p/8667045.html
測試結果如下:
說明成功安裝能使用GPU加速。
from keras.datasets import mnist from autokeras.image_supervised import ImageClassifier if __name__ == '__main__': (x_train, y_train), (x_test, y_test) = mnist.load_data(r'E:\xxxx\python_basics\自動深度學習\data\mnist.npz') x_train = x_train.reshape(x_train.shape + (1,)) x_test = x_test.reshape(x_test.shape + (1,)) clf = ImageClassifier(verbose=True) clf.fit(x_train, y_train, time_limit=1*60) clf.final_fit(x_train, y_train, x_test, y_test, retrain=True) y = clf.evaluate(x_test, y_test) print(y)