updating Dockerfile and pyproject.toml to use more recent versions of…

… pytorch and timm

	modified:   Dockerfile
	modified:   docker-compose.yml
	modified:   pyproject.toml

Dockerfile

@@ -1,4 +1,4 @@
-FROM pytorch/pytorch:2.0.1-cuda11.7-cudnn8-runtime
+FROM nvcr.io/nvidia/pytorch:24.06-py3
 
 ENV DEBIAN_FRONTEND=noninteractive
 
@@ -10,14 +10,17 @@ RUN apt-get update && apt-get install -y \
     libgl1-mesa-glx
 
 
+
 ADD torchsig/ /build/torchsig
 
 ADD pyproject.toml /build/pyproject.toml
 
 RUN pip3 install -e /build
 
-RUN pip3 install notebook jupyterlab==4.2.1
+RUN pip3 install notebook jupyterlab==4.2.3
 RUN pip3 install jupyterlab_theme_solarized_dark
 RUN pip3 install ipywidgets
 
 WORKDIR /workspace/code
+
+ADD examples/ /workspace/code/examples

docker-compose.yml

@@ -2,7 +2,7 @@ name: torch_sig_container_${PROJECT_NAME}
 services:
     torchsig_service:
         build: .
-        image: torchsig:v0.5.0
+        image: torchsig_github:v0.5.0
         container_name: torchsig_${PROJECT_NAME}
         stdin_open: true
         tty: true

pyproject.toml

@@ -5,22 +5,23 @@ build-backend = "setuptools.build_meta"
 [project]
 name = "torchsig"
 description = "Signal Processing Machine Learning Toolkit"
-authors = [{ name = "TorchSig Team" }]
+authors = [
+    {name = "TorchSig Team"},
+]
 readme = "README.md"
 requires-python = ">=3.8"
-license = { text = "MIT" }
+license = {text = "MIT"}
 classifiers = [
     "License :: OSI Approved :: MIT License",
     "Programming Language :: Python",
     "Programming Language :: Python :: 3",
 ]
 keywords = ["signal processing", "machine learning"]
 dependencies = [
-    "torch==2.0.1",
+    "torch==2.3.0",
     "torchvision",
-    "opencv-python",
-    "torchmetrics[detection]",
     "tqdm",
+    "opencv-python==4.8.0.74",
     "numpy",
     "scipy",
     "h5py",
@@ -33,16 +34,17 @@ dependencies = [
     "scikit-learn",
     "gdown",
     "icecream",
-    "timm==0.5.4",
+    "timm==1.0.7",
     "pytorch_lightning",
     "sympy",
     "numba",
-    "click",
+    "torchmetrics",
+    "click"
 ]
 dynamic = ["version"]
 
 [tool.setuptools.dynamic]
-version = { attr = "torchsig.__version__" }
+version = {attr = "torchsig.__version__"}
 
 # [project.optional-dependencies]
 # pdf = ["ReportLab>=1.2", "RXP"]