I am not able to see cu117 there

When the package installation is done in the task

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.

I think I got it resolved

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

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

thanks for the help though!!

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