Alexnet_pytorch_單GPU實現版本

純小白，純記錄 一、環境 ubuntu 18.04 CUDA 9.0 Cudnn 7.0 Opencv conda3 pycharm pytorch 二、簡介 使用Alexnet 網絡，識別貓狗圖片的分類。機子性能原因，只使用了22張圖片，epoch 只迭代了10次，只實現了訓練代碼，純學習pytorch，Alexnet用的。 三、網絡結構圖 四、代碼 代碼和數據：https://download

>>阅读原文<<