yolov12的cuda/cudnn/torch(torch-2.2.2+cu121-cp38-cp38-win_amd64.whl)/torchvision环境全安装包

最近yolov12已出,需要根据requirements.txt更新环境,此处通过百度云盘将对应文件存档。文件包括cuda_12.1.0_531.14_windows.exe、cudnn-windows-x86_64-8.9.7.29_cuda12-archive.zip、torch-2.2.2+cu121-cp38-cp38-win_amd64.whl、torchvision-0.17.2+cu121-cp38-cp38-win_amd64和一个用于测试是否安装成功的py文件。

通过网盘分享的文件:cuda12.1+cudnn8+torch2.2.2等5个文件
链接: https://pan.baidu.com/s/1Fkwa96DOC5t2q5vrq_nkyA?pwd=gqt7 提取码: gqt7
yolov12官方地址:https://github.com/sunsmarterjie/yolov12

requirements.txt

torch==2.2.2 
torchvision==0.17.2
flash_attn-2.7.3+cu11torch2.2cxx11abiFALSE-cp311-cp311-linux_x86_64.whl
timm==1.0.14
albumentations==2.0.4
onnx==1.14.0
onnxruntime==1.15.1
pycocotools==2.0.7
PyYAML==6.0.1
scipy==1.13.0
onnxslim==0.1.31
onnxruntime-gpu==1.18.0
gradio==4.44.1
opencv-python==4.9.0.80
psutil==5.9.8
py-cpuinfo==9.0.0
huggingface-hub==0.23.2
safetensors==0.4.3
numpy==1.26.4

安装记录:
1、打开控制面板卸载现有nvidia图形驱动程序,卸载后需要重新启动电脑;
2、右键点击cuda_12.1.0_531.14_windows.exe,以管理员身份运行,安装过程中一路默认;
3、解压cudnn-windows-x86_64-8.9.7.29_cuda12-archive.zip,将三个文件夹复制到C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v12.1路径下;
4、命令行pip 安装 torch-2.2.2+cu121-cp38-cp38-win_amd64.whl;
5、命令行pip 安装 torchvision-0.17.2+cu121-cp38-cp38-win_amd64;
6、命令行运行 python test.py,显示如下提示,则顺利安装。
在这里插入图片描述

深度学习pytorch之22种损失函数数学公式和代码定义
深度学习pytorch之19种优化算法(optimizer)解析
深度学习pytorch之4种归一化方法(Normalization)原理公式解析和参数使用
深度学习pytorch之简单方法自定义多种卷积即插即用
用自己的数据训练yolov11目标检测
实时语义分割之BiSeNetv2(2020)结构原理解析及建筑物提取实践
实时语义分割之Deep Dual-resolution Networks(DDRNet2021)原理解析及建筑物提取实践
SegFormer 模型解析:结合 Transformer 的高效分割框架原理及滑坡语义分割实战

<think>好的,我现在需要帮助用户找到与PyTorch 2.2.2+cu121兼容的torchvision版本。首先,我应该回忆一下PyTorchtorchvision之间的版本对应关系。通常,PyTorch官网会提供不同版本的torchtorchvision的兼容性列表。用户提到的是CUDA 12.1,对应的PyTorch版本是2.2.2,所以我要确定torchvision的哪个版本是官方推荐的。 接下来,我应该检查PyTorch的官方安装指令。比如,PyTorch官网的“Previous PyTorch Versions”页面通常会列出每个版本对应的torchvisiontorchaudio等配套库的版本。用户可能没有直接访问官网,所以我需要凭借记忆或已有的知识库来回答。例如,PyTorch 2.x版本通常对应torchvision 0.x版本,且主版本号会对应。例如,PyTorch 2.2可能对应torchvision 0.17。 另外,用户提到CUDA 12.1,这需要确认torchvision是否也支持相同的CUDA版本。通常,PyTorchtorchvisionCUDA版本必须严格匹配,否则可能出现兼容性问题。因此,正确的torchvision版本应该是编译时针对cu121的,并且版本号与PyTorch 2.2.2兼容。 然后,我需要验证这个信息是否正确。比如,查阅已有的资料或内部知识库中的相关信息。例如,引用中的内容虽然主要关于CUDA版本兼容性和其他框架安装,但可能没有直接提到torchvision版本。不过,根据一般经验,PyTorchtorchvision的版本对应是固定的,所以可以推荐用户使用PyTorch官网提供的命令来安装正确的版本。 最后,考虑到用户可能需要安装命令,我应该提供具体的pip或conda安装指令,确保用户能够正确安装兼容的torchvision版本。例如,使用pip安装时指定版本和CUDA版本,如`pip install torchvision==0.17.2+cu121`,但需要确认版本号是否正确。或者引导用户去官网查找对应命令,确保准确性。</think>根据PyTorch官方版本兼容性要求,torch 2.2.2+cu121对应的torchvision版本是0.17.2+cu121。以下是具体信息: 1. **版本对应关系** - PyTorch主版本与torchvision保持同步更新 - 当PyTorch=2.2.x时,对应torchvision=0.17.x - 必须严格匹配CUDA工具链版本(此处需要cu1212. **安装方式** ```bash # 使用pip安装 pip install torchvision==0.17.2+cu121 --extra-index-url https://download.pytorch.org/whl/cu121 # 使用conda安装 conda install pytorchvision==0.17.2 -c pytorch ``` 3. **验证安装** ```python import torch import torchvision print(torch.__version__) # 应输出2.2.2+cu121 print(torchvision.__version__) # 应输出0.17.2+cu121 ``` 建议通过PyTorch官方渠道获取安装命令,避免第三方源导致的依赖冲突[^1]。如需其他组件(如torchaudio),也需要保持相同的CUDA版本约束[^2]。
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值