This points to the wrong cu117 / driver - could that be?

I am not able to see cu117 there

So I am running a pipeline on a GCP VM, my VM has 1 NVIDIA GPU, and my requirements.txt has torch==1.13.1+cu117
torchvision==0.14.1+cu117

When I am running the Yolo training step I am getting the above error.

Sen the full Task log, you can DM it if it is easier

When the package installation is done in the task

thanks for the help though!!

Hi @<1610083503607648256:profile|DiminutiveToad80>
This sounds like the wrong container ? I think we need some more context here

I think I got it resolved