Closed RuixiangZhao closed 2 years ago
addict==2.2.1 apex==0.9.10dev einops==0.3.0 h5py==2.10.0 numpy==1.20.0 nystrom_attention==0.0.9 opencv_python==4.2.0.34 pandas==1.2.3 pytorch_lightning==1.2.3 pytorch_toolbelt==0.4.0 PyYAML==6.0 scikit_learn==1.0.2 scipy==1.5.4 setuptools==58.0.4 tensorboard==2.7.0 timm==0.3.2 torch==1.7.1+cu110 torchmetrics==0.4.1 torchvision==0.8.2+cu110 tqdm==4.46.1
same problem, have you solved it?
addict==2.2.1 apex==0.9.10dev einops==0.3.0 h5py==2.10.0 numpy==1.20.0 nystrom_attention==0.0.9 opencv_python==4.2.0.34 pandas==1.2.3 pytorch_lightning==1.2.3 pytorch_toolbelt==0.4.0 PyYAML==6.0 scikit_learn==1.0.2 scipy==1.5.4 setuptools==58.0.4 tensorboard==2.7.0 timm==0.3.2 torch==1.7.1+cu110 torchmetrics==0.4.1 torchvision==0.8.2+cu110 tqdm==4.46.1
Can you tell me your pytorch and python version?
conda create -n pytorch python=3.7 pip3 install torch==1.7.1+cu110 torchvision==0.8.2+cu110 -f https://download.pytorch.org/whl/cu110/torch_stable.html
You could simply comment out this argument amp_level
.
Hi, thank you for your sharing!
I met some problems when I tried to use multi gpus to train, which was seemly caused by pytorch-lightning version:
Can you tell me your pytorch-lightning and some other packages' version? Thank you very much!