cuda torch
参考链接：https://blog.csdn.net/weixin_42069606/article/details/105198845?utm_medium=distribute.wap_relevant.none-task-blog-BlogCommendFromMachineLearnPai2-4.wap_blog_relevant_pic&dist_request_id=1328754.7793.16171591100125415&depth_1-utm_source=distribute.wap_relevant.none-task-blog-BlogCommendFromMachineLearnPai2-4.wap_blog_relevant_pic
torch torchvison
参考链接：https://blog.csdn.net/qq_40263477/article/details/106577790