Pytorch环境搭建（吐血~）、将CPU版本换为GPU版本_yolo cpu改成gpu

作者：神奇cpp | 2024-07-19 12:19:00

踩

yolo cpu改成gpu

查看自己电脑的驱动版本。cmd命令行中输入下面代码

nvidia-smi

配置环境可能出现问题

假如报上述错误，说明对应文件夹里面没有yolov7.pt文件，与报错内容无关（因为YOLOv7官网压缩包里面没有pt文件，所以会自动联网下载，而又因为上不了外网，所以就会报错）

安装完成后的往往都是cpu版本的

对应版本好安装上xxx+cuxxx版本的，也就是GPU版本的

再次运行就可以发现是gpu版本的啦

声明：本文内容由网友自发贡献，不代表【wpsshop博客】立场，版权归原作者所有，本站不承担相应法律责任。如您发现有侵权的内容，请联系我们。转载请注明出处：https://www.wpsshop.cn/w/神奇cpp/article/detail/851455