Add Pytorch example

[karolzlot]

It may be a little tricky, but I think should be possible, it can probably be based on this nvidia container:

[karolzlot]

Oh, this is probably what I was looking for, so it may not be needed here

https://github.com/Simbotic/rust-gst-pytorch-gpu

[Chuxel]

Thanks for the suggestion! Happy to take a PR here. It would also be good to understand whether this container would only work on Linux since we try to support all three major platforms with definitions in this repository (Linux, macOS, Windows).

[karolzlot]

I see it will first need GPU support in WSL2:

NVIDIA/nvidia-docker#429 (comment)

[twsl]

GPU Support in WSL2 on Win11 is supported. I can provide a PR for a cuda-based python container im currently using.

[twsl]

@Chuxel Any chance to revisit this issue? My PR is still open.

[mimoralea]

Can you share @twsl?

[twsl]

What do you mean with share, @mimoralea ?
The PR is linked to this issue already and can be found here

[mimoralea]

Sorry, @twsl. I meant exactly that. Sorry, I'm new to VS Code developing on containers.

I tried using a Dockerfile you have in one of your repos with the "FROM ... cuda" but the build fail locally. I believe this one inherits from a different image, so will try this.

Thanks again.