We're currently not using gpu because of docker, we should implement that
We're currently not using gpu because of docker, we should implement that