Fix CUDA_VERSION env var conflict

[rsbowman-striveworks]

Description

CUDA_VERSION may be set in the container to the dot separated CUDA version number. This conflicts with the docker build arg CUDA_VERSION used to select which CUDA version to use and results in an error like this when building the image:

....
[1/2] STEP 14/15: RUN     if echo "$BASE_IMAGE" | grep -q "cuda:"; then         if [ "$CUDA_VERSION" ]; then             python ./ts_scripts/install_dependencies.py --cuda $CUDA_VERSION;         else             python ./ts_scripts/install_dependencies.py;         fi;     else         python ./ts_scripts/install_dependencies.py;     fi
usage: install_dependencies.py [-h]
                               [--cuda {cu92,cu101,cu102,cu111,cu113,cu116,cu117,cu118,cu121}]
                               [--neuronx] [--environment {prod,dev}]
                               [--nightly_torch] [--force]
install_dependencies.py: error: argument --cuda: invalid choice: '12.1.1' (choose from 'cu92', 'cu101', 'cu102', 'cu111', 'cu113', 'cu116', 'cu117', 'cu118', 'cu121')
Error: building at STEP "RUN if echo "$BASE_IMAGE" | grep -q "cuda:"; then         if [ "$CUDA_VERSION" ]; then             python ./ts_scripts/install_dependencies.py --cuda $CUDA_VERSION;         else             python ./ts_scripts/install_dependencies.py;         fi;     else         python ./ts_scripts/install_dependencies.py;     fi": while running runtime: exit status 2

Rename the docker build arg to USE_CUDA_VERSION to avoid the conflict.

Type of change

Please delete options that are not relevant.

• Bug fix (non-breaking change which fixes an issue)

Feature/Issue validation/testing

This was tested by building several images with the build_image.sh script.

Checklist:

• Did you have fun?
• Have you added tests that prove your fix is effective or that this feature works?
• Has code been commented, particularly in hard-to-understand areas?
• Have you made corresponding changes to the documentation?

[agunapal]

Interesting find. Thanks for the PR. LGTM