vscode-dev-containers icon indicating copy to clipboard operation
vscode-dev-containers copied to clipboard

Published 20 hours ago • verified publisher icon microsoft

Add Pytorch example

Open karolzlot opened this issue 5 years ago • 8 comments

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

karolzlot avatar Jan 05 '20 20:01 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

karolzlot avatar Jan 05 '20 20:01 karolzlot

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).

Chuxel avatar Jan 06 '20 21:01 Chuxel

I see it will first need GPU support in WSL2:

https://github.com/NVIDIA/nvidia-docker/issues/429#issuecomment-557264967

karolzlot avatar Jan 07 '20 17:01 karolzlot

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

twsl avatar Nov 28 '21 16:11 twsl

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

twsl avatar Feb 01 '22 02:02 twsl

Can you share @twsl?

mimoralea avatar Apr 04 '22 01:04 mimoralea

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

twsl avatar Apr 04 '22 07:04 twsl

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.

mimoralea avatar Apr 04 '22 13:04 mimoralea