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

Pytorch环境搭建(吐血~)、将CPU版本换为GPU版本_第1张图片

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

nvidia-smi 

 Pytorch环境搭建(吐血~)、将CPU版本换为GPU版本_第2张图片

配置环境可能出现问题

Pytorch环境搭建(吐血~)、将CPU版本换为GPU版本_第3张图片

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

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

Pytorch环境搭建(吐血~)、将CPU版本换为GPU版本_第4张图片

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

Pytorch环境搭建(吐血~)、将CPU版本换为GPU版本_第5张图片

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

你可能感兴趣的:(pytorch,人工智能,python)