Change docker build dynamic param to image instead of cuda version

Signed-off-by: Marcel Coetzee <marcel@mooncoon.com>