install cuda gpu version, cudann, numpy, torch