Possible to add CUDA support to the docker image?

[feltz916]

its a long an tedious process :(

[absadiki]

I think I've used the cuda-pytorch docker image to build, so cuda should be supported. Doesn't work on your end ?

[feltz916]

I cannot get the GPU option to be available. I've tried docker desktop for windows with WSL2, and uninstalled that and tried docker integrated into my WSL ubutnu 20.04. In both cases, only None and CPU are options.

[feltz916]

[absadiki]

@feltz916, I tried to add the gpu to the docker file.
Pull the changes and give it a try now ?

[feltz916]

Thanks! Its now showing up.