I ran into the same problem with a GPU which supports only compute capabilty 3.0, for me the solution was to install a legacy binary build (pytorch-0.3.0) provided by peterjc123 in his Google Drive folder.
I ran into the same problem with a GPU which supports only compute capabilty 3.0, for me the solution was to install a legacy binary build (pytorch-0.3.0) provided by peterjc123 in his Google Drive folder.