Skip to content

Commit

Permalink
updating Dockerfile and pyproject.toml to use more recent versions of…
Browse files Browse the repository at this point in the history
… pytorch and timm

	modified:   Dockerfile
	modified:   docker-compose.yml
	modified:   pyproject.toml
  • Loading branch information
pvallance committed Jul 9, 2024
1 parent d8f32f9 commit c8b5013
Show file tree
Hide file tree
Showing 3 changed files with 16 additions and 11 deletions.
7 changes: 5 additions & 2 deletions Dockerfile
Original file line number Diff line number Diff line change
@@ -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

Expand All @@ -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
2 changes: 1 addition & 1 deletion docker-compose.yml
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down
18 changes: 10 additions & 8 deletions pyproject.toml
Original file line number Diff line number Diff line change
Expand Up @@ -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",
Expand All @@ -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"]
Expand Down

0 comments on commit c8b5013

Please sign in to comment.