yolov8pytorch版本
时间: 2023-12-04 18:04:41 浏览: 138
yolov8(2023年8月版本),已经下好yolov8s.pt和yolov8n.pt
5星 · 资源好评率100%
根据引用内容,yolov8需要安装pytorch==1.8.0版本,并且需要与对应的cuda版本兼容。可以通过以下步骤来查询pytorch对应的cuda版本:
1. 创建yolov8的conda环境:
```shell
conda create -n yolov8 python=3.8
```
2. 激活yolov8环境:
```shell
conda activate yolov8
```
3. 安装pytorch==1.8.0:
```shell
conda install pytorch==1.8.0 torchvision==0.9.0 torchaudio==0.8.0 cudatoolkit=<对应的cuda版本号> -c pytorch
```
其中,<对应的cuda版本号>需要替换为你所安装的cuda版本号,例如cudatoolkit=11.1。
4. 查询pytorch对应的cuda版本:
```shell
python -c "import torch; print(torch.version.cuda)"
```
执行以上命令后,即可查询到pytorch对应的cuda版本。
阅读全文