Adding new/updated copick_generate-torch-basic-features_0.0.7

cellcanvas · Oct 22, 2024 · d52b381 · d52b381
1 parent 1078358
commit d52b381
Show file tree

Hide file tree

Showing 4 changed files with 10 additions and 4 deletions.
diff --git a/album_catalog_index.db b/album_catalog_index.db
diff --git a/solutions/copick/generate-torch-basic-features/CHANGELOG.md b/solutions/copick/generate-torch-basic-features/CHANGELOG.md
@@ -4,6 +4,9 @@ All notable changes to this project will be documented in this file.
 The format is based on [Keep a Changelog](https://keepachangelog.com/en/1.0.0/),
 and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0.html).
 
+## [0.0.7] - 2024-10-22
+More cuda deps
+
 ## [0.0.6] - 2024-10-22
 Fix dimension handling
 

diff --git a/solutions/copick/generate-torch-basic-features/solution.py b/solutions/copick/generate-torch-basic-features/solution.py
@@ -5,11 +5,14 @@
 env_file = """
 channels:
   - pytorch
+  - nvidia
   - conda-forge
 dependencies:
   - python=3.10
   - pip
   - pytorch
+  - cudatoolkit
+  - pytorch-cuda
   - torchvision
   - torchaudio
   - zarr
@@ -216,7 +219,7 @@ def compute_features(chunk_tensor, sigma_min, sigma_max, num_sigma, intensity, e
 setup(
     group="copick",
     name="generate-torch-basic-features",
-    version="0.0.6",
+    version="0.0.7",
     title="Generate Multiscale Basic Features with Torch using Copick API (Chunked, Corrected)",
     description="Compute multiscale basic features of a tomogram from a Copick run in chunks and save them using Copick's API.",
     solution_creators=["Kyle Harrington"],

diff --git a/solutions/copick/generate-torch-basic-features/solution.yml b/solutions/copick/generate-torch-basic-features/solution.yml
@@ -51,7 +51,7 @@ args:
   name: num_sigma
   required: false
   type: integer
-changelog: Fix dimension handling
+changelog: More cuda deps
 description: Compute multiscale basic features of a tomogram from a Copick run in
   chunks and save them using Copick's API.
 group: copick
@@ -64,6 +64,6 @@ tags:
 - image processing
 - cryoet
 - tomogram
-timestamp: '2024-10-22T11:27:41.823517'
+timestamp: '2024-10-22T11:45:06.794711'
 title: Generate Multiscale Basic Features with Torch using Copick API (Chunked, Corrected)
-version: 0.0.6
+version: 0.0.7