Update Dockerfile

Dockerfile

@@ -1,15 +1,12 @@
-# 1. Start from the PyTorch DEVEL image. This is the correct, full-featured base.
-FROM pytorch/pytorch:2.3.0-cuda12.1-cudnn8-devel
+# 1. Start from a community-vetted, GPU-ready image
+FROM runpod/pytorch:2.2.0-py3.10-cuda12.1.1-devel
 
-# 2. Set environment variables. The LD_LIBRARY_PATH is good practice but may be redundant
-# in this well-configured image. It doesn't hurt to keep it.
+# 2. Set up the environment
 ENV DEBIAN_FRONTEND=noninteractive
 ENV HF_HOME="/data/huggingface"
 ENV UV_CACHE_DIR="/data/uv_cache"
-ENV LD_LIBRARY_PATH=/usr/local/nvidia/lib64:$LD_LIBRARY_PATH
 
-# 3. Install git and uv. The base image has most tools, but we add these.
-RUN apt-get update && apt-get install -y --no-install-recommends git && rm -rf /var/lib/apt/lists/*
+# 3. The base image already has python, pip, etc. Install uv.
 RUN pip install uv
 
 # 4. Copy requirements and install them.