I owned an RTX3090, but I have been using Google Colab for some time for fear of the complexities of setting up an environment (especially in China, due to the GFW problem).

I finally made it, and here is the Docker compose file (running this in Win10 WSL2 docker):

services:
  jupyter:
    container_name: jupyter
    image: tensorflow/tensorflow:latest-gpu-jupyter
    shm_size: '8gb'
    deploy:
      resources:
        reservations:
          devices:
            - driver: nvidia
              count: all
              capabilities: [gpu]
    ports:
      - 8888:8888
    restart: always
    volumes:
      - type: bind
        source: /c/notebooks/jupyter
        target: /tf
    environment:
      - HTTP_PROXY=http://proxy:7890
      - HTTPS_PROXY=http://proxy:7890
      - JUPYTER_TOKEN=your-token
  proxy:
    container_name: proxy
    image: kevinwang15/convert-subscription-to-http-proxy:latest
    restart: always
    environment:
      - SUB_URL=your-ss-sub-url
  filebrowser:
    container_name: filebrowser
    image: hurlenko/filebrowser
    restart: always
    ports:
      - 3546:8080
    volumes:
      - type: bind
        source: /c/notebooks/jupyter
        target: /data
      - type: bind
        source: /c/notebooks/filebrowser/config
        target: /config
    environment:
      - FB_BASEURL=/filebrowser