docker: Make CUDA_VERSION configurable (#48199)
Summary:
makes CUDA_VERSION configurable for the docker images:
make CUDA_VERSION=10.2 CUDNN_VERSION=7 official-image
Signed-off-by: Eli Uriegas <eliuriegas@fb.com>
Pull Request resolved: https://github.com/pytorch/pytorch/pull/48199
Reviewed By: xuzhao9, janeyx99
Differential Revision: D25064256
Pulled By: seemethere
fbshipit-source-id: 25f52185097be647d11b5324f9f97cd41cdad75b
