UserWarning: Failed to load image Python extension: warn(f“Failed to load image Python extension:

当使用torchvision库时遇到运行错误,可能是torch和torchvision版本不兼容。解决方法包括:访问pytorch官网选择匹配CUDA版本的pytorch进行安装,或者查看CUDA版本后手动安装适合的torchvision版本。确保在正确的虚拟环境中执行安装命令。
摘要由CSDN通过智能技术生成

问题描述

使用torchvision库时,报错无法正常运行

问题原因和解决方式

torch和torchvision版本不匹配

1.按照官网安装对应cuda版本的pytorch

  1. 进入pytorch官网
    https://pytorch.org/
  2. 选择你的cuda版本
    在这里插入图片描述
*不知道cuda版本就执行 nvcc -V查看版本
  1. 复制红框里的指令去你的虚拟环境里安装

    *也就是说可能你需要提前执行 conda activate your-env-name

2.手动安装匹配版本

  1. 打开tochvision文档

https://pypi.org/project/torchvision/

  1. 确定自己需要安装的版本
    在这里插入图片描述
  2. 根据自己对应版本进行安装
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值