envs\pytorch-gpu\lib\site-packages\torchvision\io\image.py:13: UserWarning: Failed to load image

当遇到Pytorch和Torchvision版本不匹配时,可以按照官方GitHub上的指导进行操作。具体步骤包括在AnacondaPrompt中进入Python环境,使用pipinstall命令安装特定版本的torchvision,如pipinstalltorchvision==0.12.0-ihttp://pypi.douban.com/simple/--trusted-hostpypi.douban.com,其中版本号应替换为所需的版本。完成安装后,重启VSCODE以使更改生效。
部署运行你感兴趣的模型镜像

此问题的核心是pytorch和torchvision的版本不对应

pytorch,torchvision,python 三者的对应关系来源于官方 github,链接如下:https://github.com/pytorch/vision#installation
在这里插入图片描述

我的解决方法如下:

  1. 打开Anaconda Prompt,进入python编辑器所在的虚拟环境
  2. 输入以下内容:
pip install torchvision==0.12.0 -i http://pypi.douban.com/simple/ --trusted-host pypi.douban.com

其中0.12.0替换为上图中对应的内容即可
3. 重启VSCODE

您可能感兴趣的与本文相关的镜像

PyTorch 2.5

PyTorch 2.5

PyTorch
Cuda

PyTorch 是一个开源的 Python 机器学习库,基于 Torch 库,底层由 C++ 实现,应用于人工智能领域,如计算机视觉和自然语言处理

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值