Add cuda 10.1 support

[rongou]

Still waiting for the docker image to land, but all the tests passed locally.

This requires CUDA V10.1.168.

@hcho3 @RAMitchell

[hcho3]

@rongou Is this done?

[rongou]

@hcho3 hmm this is weird. :) The nvidia/cuda images on docker hub haven't been updated, but I do see the CUDA compiler used is the newer 10.1.168. Do you know how it got in there? Are you using some cached image on ECR?

[hcho3]

@rongou See build #2 for the build log for the 10.1 container. It looks like yum -y update command is upgrading CUDA to 10.1.168:

================================================================================
Package                       Arch   Version                     Repository
                                                                          Size
================================================================================
Updating:
binutils                      x86_64 2.20.51.0.2-5.48.el6_10.1   updates 2.8 M
cuda-command-line-tools-10-1  x86_64 10.1.168-1                  cuda    6.2 k
cuda-compat-10-1              x86_64 1:418.67-1                  cuda    7.6 M
cuda-compiler-10-1            x86_64 10.1.168-1                  cuda    6.1 k
cuda-cudart-10-1              x86_64 10.1.168-1                  cuda    148 k
cuda-cudart-dev-10-1          x86_64 10.1.168-1                  cuda    796 k
cuda-cufft-10-1               x86_64 10.1.168-1                  cuda    101 M
cuda-cufft-dev-10-1           x86_64 10.1.168-1                  cuda    193 M
cuda-cuobjdump-10-1           x86_64 10.1.168-1                  cuda     98 k
cuda-cupti-10-1               x86_64 10.1.168-1                  cuda     14 M
cuda-curand-10-1              x86_64 10.1.168-1                  cuda     42 M
cuda-curand-dev-10-1          x86_64 10.1.168-1                  cuda     65 M
cuda-cusolver-10-1            x86_64 10.1.168-1                  cuda     71 M
cuda-cusolver-dev-10-1        x86_64 10.1.168-1                  cuda     29 M
cuda-cusparse-10-1            x86_64 10.1.168-1                  cuda     77 M
cuda-cusparse-dev-10-1        x86_64 10.1.168-1                  cuda    154 M
cuda-driver-dev-10-1          x86_64 10.1.168-1                  cuda     21 k
cuda-gdb-10-1                 x86_64 10.1.168-1                  cuda    3.5 M
cuda-gpu-library-advisor-10-1 x86_64 10.1.168-1                  cuda    1.0 M
cuda-libraries-10-1           x86_64 10.1.168-1                  cuda    6.3 k
cuda-libraries-dev-10-1       x86_64 10.1.168-1                  cuda    6.4 k
cuda-license-10-1             x86_64 10.1.168-1                  cuda     22 k
cuda-memcheck-10-1            x86_64 10.1.168-1                  cuda    180 k
cuda-minimal-build-10-1       x86_64 10.1.168-1                  cuda    6.1 k
cuda-misc-headers-10-1        x86_64 10.1.168-1                  cuda    1.2 M
cuda-npp-10-1                 x86_64 10.1.168-1                  cuda     81 M
cuda-npp-dev-10-1             x86_64 10.1.168-1                  cuda     81 M
cuda-nvcc-10-1                x86_64 10.1.168-1                  cuda     23 M
cuda-nvdisasm-10-1            x86_64 10.1.168-1                  cuda     21 M
cuda-nvgraph-10-1             x86_64 10.1.168-1                  cuda     91 M
cuda-nvgraph-dev-10-1         x86_64 10.1.168-1                  cuda     55 M
cuda-nvjpeg-10-1              x86_64 10.1.168-1                  cuda    1.6 M
cuda-nvjpeg-dev-10-1          x86_64 10.1.168-1                  cuda    1.6 M
cuda-nvml-dev-10-1            x86_64 10.1.168-1                  cuda     70 k
cuda-nvprof-10-1              x86_64 10.1.168-1                  cuda    9.0 M
cuda-nvprune-10-1             x86_64 10.1.168-1                  cuda     46 k
cuda-nvrtc-10-1               x86_64 10.1.168-1                  cuda    8.8 M
cuda-nvrtc-dev-10-1           x86_64 10.1.168-1                  cuda     15 k
cuda-nvtx-10-1                x86_64 10.1.168-1                  cuda     68 k
cuda-sanitizer-api-10-1       x86_64 10.1.168-1                  cuda    3.4 M

So I suppose we are all good then?

[rongou]

@hcho3 ah that makes sense. Yeah I think we are good to go.

[hcho3]

Thanks!

[rongou]

@hcho3 BTW the images on docker hub have been updated, so for the next build, yum update should be a no-op for CUDA.

[hcho3]

Thanks for the update!