Add env for cpu

README.md

@@ -0,0 +1,15 @@
+# DINOv2 for cell classification
+
+This project is a fork of the DINOv2 repo published by meta, aiming to use the methodology presented in their papers to train a series of
+model on blood white cells images.  
+Developping a foundation model for blood white cells is interesting for several reasons:
+
+- The categories of blood white cells are not unanimous, and hematologists / datasets make different classes.
+- Some blood white cells present mutations that are visible on images, and those distinguishible features could be embedded by the model
+
+## Installing 
+
+
+## Training
+
+Most of the code used for the training 

conda.yaml

@@ -1,22 +1,23 @@
-name: dinov2
+name: cell_sim
 channels:
-  - defaults
   - pytorch
   - nvidia
-  - xformers
   - conda-forge
+  - defaults
 dependencies:
-  - python=3.9
+  - python=3.10
-  - pytorch::pytorch=2.0.0
+  - xformers>=0.0.18
-  - pytorch::pytorch-cuda=11.7.0
+  - torchmetrics
-  - pytorch::torchvision=0.15.0
+  - torchdata
   - omegaconf
-  - torchmetrics=0.10.3
+  - submitit
   - fvcore
   - iopath
-  - xformers::xformers=0.0.18
   - pip
   - pip:
-    - git+https://github.com/facebookincubator/submitit
+      - --index-url https://download.pytorch.org/whl/cu118
-    - --extra-index-url https://pypi.nvidia.com
+      - torch==2.3.0+cu118
-    - cuml-cu11
+      - torchvision
+      - --extra-index-url https://pypi.nvidia.com
+      - cuml-cu11
+

conda_cpu.yaml

@@ -15,9 +15,7 @@ dependencies:
   - iopath
   - pip
   - pip:
-      - --index-url https://download.pytorch.org/whl/cu118
+      - --index-url https://download.pytorch.org/whl/cpu
-      - torch==2.3.0+cu118
+      - torch==2.3.0+cpu
       - torchvision
-      - --extra-index-url https://pypi.nvidia.com
-      - cuml-cu11
 

conda_or.yaml

@@ -0,0 +1,22 @@
+name: dinov2
+channels:
+  - defaults
+  - pytorch
+  - nvidia
+  - xformers
+  - conda-forge
+dependencies:
+  - python=3.9
+  - pytorch::pytorch=2.0.0
+  - pytorch::pytorch-cuda=11.7.0
+  - pytorch::torchvision=0.15.0
+  - omegaconf
+  - torchmetrics=0.10.3
+  - fvcore
+  - iopath
+  - xformers::xformers=0.0.18
+  - pip
+  - pip:
+    - git+https://github.com/facebookincubator/submitit
+    - --extra-index-url https://pypi.nvidia.com
+    - cuml-cu11

notebooks/Untitled.ipynb

@@ -0,0 +1,6 @@
+{
+ "cells": [],
+ "metadata": {},
+ "nbformat": 4,
+ "nbformat_minor": 5
+}

requirements.txt

@@ -7,5 +7,3 @@ fvcore
 iopath
 xformers>=0.0.18
 submitit
---extra-index-url https://pypi.nvidia.com
-cuml-cu11