Pytorch安装:
# CUDA 11.6
pip install torch==1.12.1+cu116 torchvision==0.13.1+cu116 torchaudio==0.12.1 --extra-index-url https://download.pytorch.org/whl/cu116
# CUDA 11.3
pip install torch==1.12.1+cu113 torchvision==0.13.1+cu113 torchaudio==0.12.1 --extra-index-url https://download.pytorch.org/whl/cu113
# CUDA 10.2
pip install torch==1.12.1+cu102 torchvision==0.13.1+cu102 torchaudio==0.12.1 --extra-index-url https://download.pytorch.org/whl/cu102
# CPU only
pip install torch==1.12.1+cpu torchvision==0.13.1+cpu torchaudio==0.12.1 --extra-index-url https://download.pytorch.org/whl/cpu
在安装Pytorch后,直接运行main.py即可
深度学习之PyTorch实战源码：基于轻量级网络【Mobilenet V2】处理MNIST数据集，实现99%+准确率
需积分: 5 13 浏览量
2023-12-14
12:11:54
上传
评论 1
收藏 6KB ZIP 举报
mobilenet_v2.zip （10个子文件） 
mobilenet_v2 
main.py 4KB
README.txt 650B .idea mobilenet_v2.iml 291B workspace.xml 2KB misc.xml 292B inspectionProfiles Project_Default.xml 251B profiles_settings.xml 174B modules.xml 283B deployment.xml 4KB .gitignore 190B资源评论
