Skip to content
Snippets Groups Projects
Commit 65f29eed authored by André Anjos's avatar André Anjos :speech_balloon:
Browse files

[helpers/cuda] Setup pixi environment

parent ed3ebc95
No related branches found
No related tags found
1 merge request!33Full dev environment based on pixi-only
Pipeline #86347 passed
Source diff could not be displayed: it is too large. Options to address this: view the blob.
......@@ -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"
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment