Pytorch环境搭建(吐血~)、将CPU版本换为GPU版本

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

nvidia-smi 

 

配置环境可能出现问题

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

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

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

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

猜你喜欢

转载自blog.csdn.net/weixin_60535956/article/details/133222073