@shamoons @Daniel_Hernandez  could you both post the output of python -m torch.utils.collect_env?
@Daniel_Hernandez 's env shows CUDA11.0, which is not shipped in the 1.8.0 binaries (10.2 and 11.1 are used), so I’m unsure where this package comes from. Did you build it from source?
If you are using the 10.2 pip wheels (not conda), note that sm_75 was pruned from them and you would have to use the 11.1 pip wheels.
This issue is also tracked here 124.