yolov7调用gpu之torch下载
1.删除之前的troch
直接pip requirements.txt里的troch是cpu版的我们要先删除
找到解释器位置我的是d盘的jieshiqi11111然后打开lib然后site-…
然后往下滑找到torch及以torch开头的文件夹全部删除
2.查看cuda版本并下载torch gpu版(清华源毁一生)
用清华源安装基本都是cpu版
打开cmd输入
nvidia-smi
查看版本后进入官网下载Pytorch
pytorch: https://pytorch.org/get-started/locally/
然后复制下面那个命令去终端pip
3.检测是否安装成功
import torch
print(torch.cuda.is_available())
输出结果为true ok安装成功在使用yolo的时候调用gpu就行了具体自己搜索