Documentation on requirements

[shellycsy]

It would be nice if the readme included a couple of lines on requirements, specially the cuda version that this code was ran with（I'm getting an error which I'm guessing it's due to cuda version）. thank you