Add Memray, Change to PyTorch base image (#201)

Dockerfile

@@ -1,13 +1,16 @@
-FROM python:3.11
+FROM pytorch/pytorch:latest
+
+USER root
+
+RUN --mount=type=cache,target=/.cache/pip pip install --upgrade pip
+RUN --mount=type=cache,target=/.cache/pip pip install ipython memray
 
-RUN --mount=type=cache,target=/root/.cache pip install --upgrade pip
-RUN --mount=type=cache,target=/root/.cache pip install ipython
 
 RUN mkdir /app
 COPY src/ /app/src/
 COPY tox.ini pyproject.toml setup.py setup.cfg LICENSE.txt README.md /app/
 
-RUN --mount=type=cache,target=/root/.cache cd /app && pip install '.[dev,test]' --extra-index-url https://download.pytorch.org/whl/cpu
+RUN --mount=type=cache,target=/.cache/pip cd /app && pip install '.[dev,test]'
 
 ENV PYTHONUNBUFFERED=1
 

pyproject.toml

@@ -3,7 +3,7 @@ name = "bayestme"
 version = "1.0.0"
 description = "A reference-free Bayesian method that discovers spatial transcriptional programs in the tissue microenvironment"
 readme = "README.md"
-requires-python = ">=3.11"
+requires-python = ">=3.10"
 license = {file = "LICENSE.txt"}
 keywords = ["spatial", "expression", "rna"]
 authors = [