[helpers/cuda] Setup pixi environment

helpers/cuda/pixi.toml

@@ -9,23 +9,8 @@
 # of the mednet environment.  It loosely follows the recipe in the pixi
 # documentation: https://pixi.sh/latest/advanced/channel_priority/#use-case-pytorch-and-nvidia-with-conda-forge
 #
-# USAGE INSTRUCTIONS
-# ------------------
-#
-# To use this file, first link it to `pixi.toml`, and re-run `pixi install`.
-# Having a file named `pixi.toml` on the current directory has precedence over
-# `pyproject.toml`:
-#
-# $ rm -rf .pixi pixi.lock  # clean-up
-# $ ln -s helpers/cuda/pixi.toml .
-# $ ln -s helpers/cuda/pixi.lock .
-# $ pixi install  # this will install all support libraries
-#
-#
-# WHY IS THIS SEPARATED FROM THE MAIN MANIFEST
-# --------------------------------------------
-#
-# Check https://github.com/prefix-dev/pixi/issues/1051
+# Why is this separated from the main manifest? Check
+# https://github.com/prefix-dev/pixi/issues/1051.
 
 [project]
 name = "mednet"
@@ -50,11 +35,11 @@ psutil = "*"
 tabulate = "*"
 matplotlib = "*"
 pillow = "*"
-pytorch = { version = ">=2.1.2", channel = "pytorch" }
-torchvision = { version = ">=0.16.1", channel = "pytorch" }
-lightning = ">=2.2.1"
+pytorch = { version = "~=2.2.2", channel = "pytorch" }
+torchvision = { version = "~=0.17.2", channel = "pytorch" }
+lightning = "~=2.2.1"
 tensorboard = "*"
-grad-cam = ">=1.5.0"
+grad-cam = "~=1.5.0"
 versioningit = "*"
 # special stuff
 cuda = { version = "*", channel = "nvidia" }
@@ -63,6 +48,7 @@ pytorch-cuda = { version = "12.1.*", channel = "pytorch" }
 [host-dependencies]
 hatch = "*"
 versioningit = "*"
+pdbpp = "*"
 
 [system-requirements]
 cuda = "12.1"