Hi,

Thanks again for releasing the Dockerfile for code-server and PyTorch. I was trying to customize it for my own use case however I stumbled upon a problem. Before I begin, I must say that using your prebuilt image ghcr.io/works-on-my-machine/pytorch-code-server:1.9.1 works perfectly fine. It's just that when I build the docker file in this repo on my end and launch it, the connection keeps on dropping.

I am running on the image on a Linux machine which otherwise never had any problem with ssh. I tested both by port-forwarding it to my local machine (Mac OS, Chrome) as well as accessing it on the very same machine. In both cases, I am getting the same problem. Do you know what can be the problem? Can you share the original docker file you used to build original image?

Thanks!