使用YOLO时torchtorchvisioncuda版本下载
Posted weixin_46472346
tags:
篇首语：本文由小常识网(cha138.com)小编为大家整理，主要介绍了使用YOLO时torchtorchvisioncuda版本下载相关的知识，希望对你有一定的参考价值。
使用YOLO时torch、torchvision、cuda版本下载
1.根据电脑GPU型号确定CUDA版本
去英伟达官网下载，具体去别处搜.
我的GPU得到：CUDA11.1
2.先大致确定使用的python、pytorch范围
根据yolov8的要求，需要python在3.10~3.7之间，pytorch>=1.7
但是还不要下载，因为新版本的pytorch对应的torchvision可能没有适合于windows的cuda版本。
3.选择最新版的cuda版本的torchvision。再对应下载torch
https://download.pytorch.org/whl/torch_stable.html
里面首先ctrl+F找到cu111，对应CUDA11.1版本
（1）找到支持windows的最高torchvision的wheel文件，下载
（2）https://github.com/pytorch/vision
在上面找到对应的torch版本
（3）返回https://download.pytorch.org/whl/torch_stable.html
找到cu111的torch版本
2023/02/08 我选择的版本
以上是关于使用YOLO时torchtorchvisioncuda版本下载的主要内容，如果未能解决你的问题，请参考以下文章