AttributeError: module 'torch.cuda' has no attribute 'amp'

KaihuaTang / Scene-Graph-Benchmark.pytorch

A new codebase for popular Scene Graph Generation methods (2020). Visualization & Scene Graph Extraction on custom images/datasets are provided. It's also a PyTorch implementation of paper “Unbiased Scene Graph Generation from Biased Training CVPR 2020”

MIT License

1.03k stars 228 forks source link

AttributeError: module 'torch.cuda' has no attribute 'amp' #155

Open tongbaochen opened 2 years ago

tongbaochen commented 2 years ago

❓ Questions and Help

hi, I have successfully built up the Scene-Graph-Benchmark.pytorch followed the instruction of https://github.com/KaihuaTang/Scene-Graph-Benchmark.pytorch/blob/master/INSTALL.md howerver, when I'm trying to run the testing code. I have run into this issue:

[ ] AttributeError: module 'torch.cuda' has no attribute 'amp' is there anyone know how to solved it? the config is as the followings: CUDA 10.1 torch 1.4 gcc 7.5 python 3.6

JASON7033 commented 2 years ago

uncessfully compiled the apex , remove build file in apex , try again?