By using AWS re:Post, you agree to the AWS re:Post Terms of Use
AWS re:Postre:Post

Upgrade NVIDIA drivers Sagemaker notebook instances

0

Hi, Is it possible to upgrade the NVIDIA drivers on the g4dn series of instances on Sagemaker? They currently have 510 which has a max cuda of 11.6. Pytorch 2 is soon to be released and will only support >=CUDA 11.7. Without an upgrade these instances won't be able to use pytorch 2. What is the correct command to run to upgrade these?

Thanks

Topics
Machine Learning & AI
Tags
Amazon SageMakerMachine Learning & AI
Language
English
rePost-User-6380390
asked a year ago73 views
No Answers

You are not logged in. Log in to post an answer.

A good answer clearly answers the question and provides constructive feedback and encourages professional growth in the question asker.

Guidelines for Answering Questions

Relevant content