pytorch版本训练的版本咨询

[WMX1007]

我使用的是自己的数据集，执行train.py时总是报错ImportError: cannot import name 'datapoints' from 'torchvision'
AttributeError: module 'torchvision.transforms' has no attribute 'RandomPhotometricDistort'等
查了一下没有相关错误解析，不知道是不是版本原因，我是用的是2.2.0版本的torch，0.17.0版本的torchvision

[lyuwenyu]

是的 版本问题
详细参考：https://github.com/lyuwenyu/RT-DETR/blob/main/rtdetr_pytorch/requirements.txt

[a-vegatable-bird]

升级就行：conda install pytorch==2.0.1 torchvision==0.15.2 torchaudio==2.0.2 pytorch-cuda=11.7 -c pytorch -c nvidia