No, I don’t think so, as it doesn’t change the behavior of the binaries and is only used if you build PyTorch from source, which isn’t the case if I understand your workflow correctly.
The original error message is raised, if you’ve installed a pip wheel or conda binary, which doesn’t support your architecture. Based on the message:



 Ilias_Giannakopoulos:

The current PyTorch install supports CUDA capabilities sm_37 sm_50 sm_60 sm_70.


it seems you were using a pip wheel with CUDA<=10.2.