训练一轮结束后报错AttributeError: module 'torchvision' has no attribute 'ops'

已经下载conda install pytorch==2.0.0 torchvision==0.15.0 torchaudio==2.0.0 pytorch-cuda=11.8 -c pytorch -c nvidia和12.1的cuda,代码能跑一轮,但是一轮后就报错了这,之前出现过,更改版本后直接无法使用gpu跑了

更新 torchvision 版本

pip install --upgrade torchvision
  • 你可以参考下这个问题的回答, 看看是否对你有帮助, 链接: https://ask.csdn.net/questions/7586246
  • 除此之外, 这篇博客: 用conda下载安装pytorch-gpu版|torch.cuda.is_available()False|torch.cuda has no attribute中的 下载gpu版pytorch 部分也许能够解决你的问题, 你可以仔细阅读以下内容或跳转源博客中阅读:
  • 在下载之前,有个关键步骤(血泪教训)。输入以下命令,看conda下有没有cpuonly这个包,有的话卸载,否则后续默认下载的是cpu版pytorch。

    conda list
    

    在这里插入图片描述
    卸载后在pytorch官网上找到对应的命令进行下载。
    在这里插入图片描述
    因为我的conda是10.1版本,可以参照previous-versions。又一个坑!pytorch、torchvision、torchchaudio版本要匹配,卸载不干净就回滚回以前版本“conda list --revision”。
    在这里插入图片描述

    conda install pytorch==1.7.1 torchvision==0.8.2 torchaudio==0.7.2 cudatoolkit=10.1 -c pytorch