Tensorflow 如何在sagemaker培训实例中指定cuda版本_Tensorflow_Amazon Sagemaker

Tensorflow 如何在sagemaker培训实例中指定cuda版本

tensorflow

Tensorflow 如何在sagemaker培训实例中指定cuda版本,tensorflow,amazon-sagemaker,Tensorflow,Amazon Sagemaker,我目前正在使用tensorflow 2.3在AWS sagemaker上训练CNN模型。我想在GPU实例上运行培训，但是看起来默认的cuda版本是cuda 10.0，但我需要的是cuda 10.1。我需要一种在培训实例中指定cuda版本的方法。我使用estimator.Tensorflow，它使用脚本文件，而不是docker图像。和

我目前正在使用tensorflow 2.3在AWS sagemaker上训练CNN模型。我想在GPU实例上运行培训，但是看起来默认的cuda版本是cuda 10.0，但我需要的是cuda 10.1。我需要一种在培训实例中指定cuda版本的方法。我使用estimator.Tensorflow，它使用脚本文件，而不是docker图像。

和