release

mhamilton723 · Feb 27, 2024 · 0d51734 · 0d51734
1 parent 1d750f8
commit 0d51734
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/.github/workflows/build_and_upload.yml b/.github/workflows/build_and_upload.yml
@@ -9,7 +9,7 @@ jobs:
   build-and-upload:
     runs-on: ubuntu-latest
     container:
-      image: nvidia/cuda:11.0-base  # Use an appropriate CUDA image
+      image: nvidia/cuda:11.8.0-cudnn8-runtime-ubuntu22.04
       options: --gpus all
     steps:
     - uses: actions/checkout@v2

diff --git a/setup.py b/setup.py
@@ -3,7 +3,7 @@
 
 setup(
     name='featup',
-    version='0.1.1',
+    version='0.1.2',
     packages=find_packages(),
     install_requires=[
         'torch',