CUDNN_STATUS_NOT_INITIALIZED when installing pytorch with pip but not with conda

Double post from here.