Revert "CUDA incompatible"

This reverts commit 5da5d691a2aaee4a669d4f1c548dad2f6db0cded.

Dockerfile

@@ -8,7 +8,6 @@ RUN apt-get update -y && \
 	rm -rf /var/lib/apt/lists/*
 
 COPY requirements_docker.txt requirements_versions.txt /tmp/
-RUN pip cache purge
 RUN pip install --no-cache-dir -r /tmp/requirements_docker.txt -r /tmp/requirements_versions.txt && \
 	rm -f /tmp/requirements_docker.txt /tmp/requirements_versions.txt
 RUN pip install --no-cache-dir xformers==0.0.23 --no-dependencies