import torch不报错但是import torchvision报错:UserWarning:Faild to load image Python extension:Could not解决二

import torch不报错但是import torchvision报错:UserWarning:Faild to load image Python extension:Could not解决二,第1张

import torch不报错但是import torchvision报错:UserWarning:Faild to load image Python extension:Could not解决二 import torch不报错但是import torchvision报错:UserWarning:Faild to load image Python extension:Could not解决二

可能是你的torch和torchvision版本不对应

我是在运行yolo5时发现的错误,yolo5要求torch>=1.6就可以,那我们就先卸载原来的torch和torchvision,再安装新的,打开anaconda prompt,
1.输入如下命令卸载

pip uninstall torch
pip uninstall torchvision

2.然后查看对应cuda和torch1.6.0和torchvision0.7.0命令进行安装


或者想安装其他版本torch,可以在https://pytorch.org/查询
3.要查看自己cuda版本只需要cmd输入如下命令:

nvcc -V

4.我是cuda10.2

conda install pytorch==1.6.0 torchvision==0.7.0 cudatoolkit=10.2 -c pytorch

欢迎分享,转载请注明来源:内存溢出

原文地址: https://outofmemory.cn/zaji/5689825.html

(0)
打赏 微信扫一扫 微信扫一扫 支付宝扫一扫 支付宝扫一扫
上一篇 2022-12-17
下一篇 2022-12-17

发表评论

登录后才能评论

评论列表(0条)

保存