V100 compatibility

Hi, I build PyTorch from source by

TORCH_CUDA_ARCH_LIST="3.7 5.0 6.0 7.0 7.5 8.0 8.6"  python setup.py install

But why it can not run on V100 ? This pytorch version can run in A100 machine. And I notice that the RuntimeError happened in optimization step given by Apex, and this is how I install apex after pytorch

git clone https://github.com/NVIDIA/apex
cd apex
pip install -v --no-cache-dir --global-option="--cpp_ext" --global-option="--cuda_ext" ./



This seems to be relevant with Apex installation. I found this, and I am trying

I guess you haven’t specified the architectures while building apex so would need to rebuild it.

Yes ! Exactly, after specifying the architectures when building apex, it works! Thank you