torch gpu版本下载(yolo)

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就行了具体自己搜索