Where can I find the CUDA version used in the latest Amazon SageMaker Distribution?

Question

I'm using SageMaker Studio and want to know which CUDA version is used in the latest SageMaker Distribution (version 1.4) when running on a GPU instance.

I checked the Dockerfile in the public GitHub repository for SageMaker, and I noticed that the CUDA versions are [specified as ARGs](https://github.com/aws/sagemaker-distribution/blob/17300e1d4df453f34c795205568a746bc890be1f/template/v1/Dockerfile#L4). However, I couldn't find the specific ARG value used for SageMaker Distribution 1.4.

Is there any documentation or resource that provides information about the CUDA version used in SageMaker Distribution 1.4?

Is there any official documentation, release notes, or changelog that provides information about the CUDA version used in SageMaker Distribution 1.4?

Please note: I'm looking for documentation or resources that directly state the CUDA version. I'm aware of programmatic ways to check the CUDA version, so please focus answers on relevant documentation instead.

Answer

You can refer to the `gpu.env.out` file for each version. For example, cuda version for v1.4.0 is here - https://github.com/aws/sagemaker-distribution/blob/17300e1d4df453f34c795205568a746bc890be1f/build_artifacts/v1/v1.4/v1.4.0/gpu.env.out#L42

Under `build_artifacts`, you can also see changelog for each version.

Where can I find the CUDA version used in the latest Amazon SageMaker Distribution?

Contenido relevante