Update environment.yml

environment.yml

@@ -1,313 +1,20 @@
 name: cloudspace
 channels:
+  - pytorch
   - conda-forge
   - defaults
 dependencies:
-  - _libgcc_mutex=0.1=main
-  - _openmp_mutex=5.1=1_gnu
-  - aiohappyeyeballs=2.4.3=py310h06a4308_0
-  - alembic=1.13.3=py310h06a4308_0
-  - aniso8601=9.0.1=pyhd3eb1b0_0
-  - arrow-cpp=16.1.0=hc1eb8f0_0
-  - attrs=24.2.0=py310h06a4308_0
-  - bcrypt=3.2.0=py310h5eee18b_1
-  - blas=1.0=openblas
-  - blinker=1.6.2=py310h06a4308_0
-  - boost-cpp=1.82.0=hdb19cb5_2
-  - bottleneck=1.4.2=py310ha9d4c09_0
-  - brotli=1.0.9=h5eee18b_8
-  - brotli-bin=1.0.9=h5eee18b_8
-  - brotli-python=1.0.9=py310h6a678d5_8
-  - bzip2=1.0.8=h5eee18b_6
-  - c-ares=1.19.1=h5eee18b_0
-  - ca-certificates=2024.11.26=h06a4308_0
-  - certifi=2024.8.30=py310h06a4308_0
-  - cffi=1.17.1=py310h1fdaa30_0
-  - click=8.1.7=py310h06a4308_0
-  - cloudpickle=3.0.0=py310h06a4308_0
-  - contourpy=1.3.1=py310hdb19cb5_0
-  - cryptography=43.0.3=py310h7825ff9_1
-  - databricks-sdk=0.33.0=py310h06a4308_0
-  - deprecated=1.2.13=py310h06a4308_0
-  - docker-py=7.1.0=py310h06a4308_0
-  - entrypoints=0.4=py310h06a4308_0
-  - flask=3.0.3=py310h06a4308_0
-  - freetype=2.12.1=h4a9f257_0
-  - frozenlist=1.5.0=py310h5eee18b_0
-  - gflags=2.2.2=h6a678d5_1
-  - gitdb=4.0.7=pyhd3eb1b0_0
-  - gitpython=3.1.43=py310h06a4308_0
-  - glog=0.5.0=h6a678d5_1
-  - graphene=3.3=py310h06a4308_0
-  - graphql-core=3.2.3=py310h06a4308_1
-  - graphql-relay=3.2.0=py310h06a4308_0
-  - greenlet=3.0.1=py310h6a678d5_0
-  - gunicorn=22.0.0=py310h06a4308_0
-  - icu=73.1=h6a678d5_0
-  - importlib-metadata=8.5.0=py310h06a4308_0
-  - itsdangerous=2.2.0=py310h06a4308_0
-  - jinja2=3.1.4=py310h06a4308_1
-  - joblib=1.4.2=py310h06a4308_0
-  - jpeg=9e=h5eee18b_3
-  - krb5=1.20.1=h143b758_1
-  - lcms2=2.12=h3be6417_0
-  - ld_impl_linux-64=2.40=h12ee557_0
-  - lerc=3.0=h295c915_0
-  - libabseil=20240116.2=cxx17_h6a678d5_0
-  - libboost=1.82.0=h109eef0_2
-  - libbrotlicommon=1.0.9=h5eee18b_8
-  - libbrotlidec=1.0.9=h5eee18b_8
-  - libbrotlienc=1.0.9=h5eee18b_8
-  - libcurl=8.9.1=h251f7ec_0
-  - libdeflate=1.17=h5eee18b_1
-  - libedit=3.1.20230828=h5eee18b_0
-  - libev=4.33=h7f8727e_1
-  - libevent=2.1.12=hdbd6064_1
-  - libffi=3.4.4=h6a678d5_1
-  - libgcc-ng=11.2.0=h1234567_1
-  - libgfortran-ng=11.2.0=h00389a5_1
-  - libgfortran5=11.2.0=h1234567_1
-  - libgomp=11.2.0=h1234567_1
-  - libgrpc=1.62.2=h2d74bed_0
-  - libnghttp2=1.57.0=h2d74bed_0
-  - libopenblas=0.3.21=h043d6bf_0
-  - libpng=1.6.39=h5eee18b_0
-  - libprotobuf=4.25.3=he621ea3_0
-  - libsodium=1.0.18=h7b6447c_0
-  - libssh2=1.11.1=h251f7ec_0
-  - libstdcxx-ng=11.2.0=h1234567_1
-  - libthrift=0.15.0=h1795dd8_2
-  - libtiff=4.5.1=h6a678d5_0
-  - libuuid=1.41.5=h5eee18b_0
-  - libwebp-base=1.3.2=h5eee18b_1
-  - lz4-c=1.9.4=h6a678d5_1
-  - mako=1.2.3=py310h06a4308_0
-  - matplotlib-base=3.9.2=py310hbfdbfaf_1
-  - mlflow=2.18.0=hff52083_0
-  - mlflow-skinny=2.18.0=py310hff52083_0
-  - mlflow-ui=2.18.0=py310hff52083_0
-  - multidict=6.1.0=py310h5eee18b_0
-  - ncurses=6.4=h6a678d5_0
-  - numexpr=2.10.1=py310hd28fd6d_0
-  - numpy=1.26.4=py310heeff2f4_0
-  - numpy-base=1.26.4=py310h8a23956_0
-  - openjpeg=2.5.2=he7f1fd0_0
-  - openssl=3.0.15=h5eee18b_0
-  - opentelemetry-api=1.16.0=pyhd8ed1ab_0
-  - opentelemetry-sdk=1.16.0=pyhd8ed1ab_0
-  - opentelemetry-semantic-conventions=0.37b0=pyhd8ed1ab_0
-  - orc=2.0.1=h2d29ad5_0
-  - paramiko=3.5.0=py310h06a4308_0
-  - pillow=11.0.0=py310hfdbf927_0
-  - prometheus_client=0.21.0=py310h06a4308_0
-  - prometheus_flask_exporter=0.22.4=py310h06a4308_0
-  - propcache=0.2.0=py310h5eee18b_0
-  - pyarrow=16.1.0=py310h1128e8f_0
-  - pynacl=1.5.0=py310h5eee18b_0
-  - pyopenssl=24.2.1=py310h06a4308_0
-  - pyparsing=3.2.0=py310h06a4308_0
-  - pysocks=1.7.1=py310h06a4308_0
-  - python=3.10.15=he870216_1
-  - python-dateutil=2.9.0post0=py310h06a4308_2
-  - python-tzdata=2023.3=pyhd3eb1b0_0
-  - python_abi=3.10=2_cp310
-  - pyyaml=6.0.2=py310h5eee18b_0
-  - querystring_parser=1.2.4=py310h06a4308_0
-  - re2=2022.04.01=h295c915_0
-  - readline=8.2=h5eee18b_0
-  - requests=2.32.3=py310h06a4308_1
-  - s2n=1.3.27=hdbd6064_0
-  - setuptools=75.1.0=py310h06a4308_0
-  - six=1.16.0=pyhd3eb1b0_1
-  - smmap=4.0.0=pyhd3eb1b0_0
-  - snappy=1.2.1=h6a678d5_0
-  - sqlalchemy=2.0.34=py310h00e1ef3_0
-  - sqlite=3.45.3=h5eee18b_0
-  - sqlparse=0.4.4=py310h06a4308_0
-  - threadpoolctl=3.5.0=py310h2f386ee_0
-  - tk=8.6.14=h39e8969_0
-  - typing_extensions=4.11.0=py310h06a4308_0
-  - unicodedata2=15.1.0=py310h5eee18b_0
-  - urllib3=2.2.3=py310h06a4308_0
-  - utf8proc=2.6.1=h5eee18b_1
-  - websocket-client=1.8.0=py310h06a4308_0
-  - wheel=0.44.0=py310h06a4308_0
-  - wrapt=1.14.1=py310h5eee18b_0
-  - xz=5.4.6=h5eee18b_1
-  - yaml=0.2.5=h7b6447c_0
-  - yarl=1.18.0=py310h5eee18b_0
-  - zipp=3.21.0=py310h06a4308_0
-  - zlib=1.2.13=h5eee18b_1
-  - zstd=1.5.6=hc292b87_0
+  - python=3.9
+  - pytorch
+  - torchvision
+  - cudatoolkit=11.3
+  - scikit-learn
+  - matplotlib
+  - pillow
+  - numpy
+  - tqdm
+  - mlflow
+  - gradio
+  - pip
   - pip:
-      - absl-py==2.1.0
-      - aiohttp==3.11.7
-      - aiosignal==1.3.1
-      - annotated-types==0.7.0
-      - anyio==4.6.2.post1
-      - argon2-cffi==23.1.0
-      - argon2-cffi-bindings==21.2.0
-      - arrow==1.3.0
-      - asttokens==2.4.1
-      - async-lru==2.0.4
-      - async-timeout==5.0.1
-      - babel==2.16.0
-      - backoff==2.2.1
-      - beautifulsoup4==4.12.3
-      - bleach==6.2.0
-      - boto3==1.35.70
-      - botocore==1.35.70
-      - cachetools==5.5.0
-      - charset-normalizer==3.4.0
-      - comm==0.2.2
-      - cycler==0.12.1
-      - debugpy==1.8.9
-      - decorator==5.1.1
-      - defusedxml==0.7.1
-      - exceptiongroup==1.2.2
-      - executing==2.1.0
-      - fastapi==0.115.5
-      - fastjsonschema==2.20.0
-      - filelock==3.16.1
-      - fire==0.7.0
-      - fonttools==4.55.0
-      - fqdn==1.5.1
-      - fsspec==2024.10.0
-      - git-filter-repo==2.47.0
-      - google-auth==2.36.0
-      - google-auth-oauthlib==1.2.1
-      - grpcio==1.68.0
-      - h11==0.14.0
-      - httpcore==1.0.7
-      - httptools==0.6.4
-      - httpx==0.27.2
-      - idna==3.10
-      - imageio==2.36.1
-      - ipykernel==6.26.0
-      - ipython==8.17.2
-      - ipywidgets==8.1.1
-      - isoduration==20.11.0
-      - jedi==0.19.2
-      - jmespath==1.0.1
-      - json5==0.10.0
-      - jsonpointer==3.0.0
-      - jsonschema==4.23.0
-      - jsonschema-specifications==2024.10.1
-      - jupyter-client==8.6.3
-      - jupyter-core==5.7.2
-      - jupyter-events==0.10.0
-      - jupyter-lsp==2.2.5
-      - jupyter-server==2.14.2
-      - jupyter-server-terminals==0.5.3
-      - jupyterlab==4.2.0
-      - jupyterlab-pygments==0.3.0
-      - jupyterlab-server==2.27.3
-      - jupyterlab-widgets==3.0.13
-      - kiwisolver==1.4.7
-      - lazy-loader==0.4
-      - lightning==2.4.0
-      - lightning-cloud==0.5.70
-      - lightning-sdk==0.1.30
-      - lightning-utilities==0.11.9
-      - lime==0.2.0.1
-      - litdata==0.2.32
-      - litserve==0.2.5
-      - markdown==3.7
-      - markdown-it-py==3.0.0
-      - markupsafe==3.0.2
-      - matplotlib==3.8.2
-      - matplotlib-inline==0.1.7
-      - mdurl==0.1.2
-      - mistune==3.0.2
-      - mpmath==1.3.0
-      - nbclient==0.10.0
-      - nbconvert==7.16.4
-      - nbformat==5.10.4
-      - nest-asyncio==1.6.0
-      - networkx==3.4.2
-      - notebook-shim==0.2.4
-      - nvidia-cublas-cu12==12.1.3.1
-      - nvidia-cuda-cupti-cu12==12.1.105
-      - nvidia-cuda-nvrtc-cu12==12.1.105
-      - nvidia-cuda-runtime-cu12==12.1.105
-      - nvidia-cudnn-cu12==8.9.2.26
-      - nvidia-cufft-cu12==11.0.2.54
-      - nvidia-curand-cu12==10.3.2.106
-      - nvidia-cusolver-cu12==11.4.5.107
-      - nvidia-cusparse-cu12==12.1.0.106
-      - nvidia-nccl-cu12==2.19.3
-      - nvidia-nvjitlink-cu12==12.6.85
-      - nvidia-nvtx-cu12==12.1.105
-      - oauthlib==3.2.2
-      - overrides==7.7.0
-      - packaging==24.2
-      - pandas==2.1.4
-      - pandocfilters==1.5.1
-      - parso==0.8.4
-      - pexpect==4.9.0
-      - pip==24.3.1
-      - platformdirs==4.3.6
-      - prompt-toolkit==3.0.48
-      - protobuf==4.23.4
-      - psutil==6.1.0
-      - ptyprocess==0.7.0
-      - pure-eval==0.2.3
-      - pyasn1==0.6.1
-      - pyasn1-modules==0.4.1
-      - pycparser==2.22
-      - pydantic==2.10.2
-      - pydantic-core==2.27.1
-      - pygments==2.18.0
-      - pyjwt==2.10.0
-      - python-dotenv==1.0.1
-      - python-json-logger==2.0.7
-      - python-multipart==0.0.17
-      - pytorch-lightning==2.4.0
-      - pytz==2024.2
-      - pyzmq==26.2.0
-      - referencing==0.35.1
-      - requests-oauthlib==2.0.0
-      - rfc3339-validator==0.1.4
-      - rfc3986-validator==0.1.1
-      - rich==13.9.4
-      - rpds-py==0.21.0
-      - rsa==4.9
-      - s3transfer==0.10.4
-      - scikit-image==0.24.0
-      - scikit-learn==1.3.2
-      - scipy==1.11.4
-      - send2trash==1.8.3
-      - simple-term-menu==1.6.5
-      - sniffio==1.3.1
-      - soupsieve==2.6
-      - stack-data==0.6.3
-      - starlette==0.41.3
-      - sympy==1.13.3
-      - tensorboard==2.15.1
-      - tensorboard-data-server==0.7.2
-      - termcolor==2.5.0
-      - terminado==0.18.1
-      - tifffile==2024.9.20
-      - tinycss2==1.4.0
-      - tomli==2.1.0
-      - torch==2.2.1+cu121
-      - torchmetrics==1.3.1
-      - torchvision==0.17.1+cu121
-      - tornado==6.4.2
-      - tqdm==4.67.1
-      - traitlets==5.14.3
-      - triton==2.2.0
-      - types-python-dateutil==2.9.0.20241003
-      - typing-extensions==4.12.2
-      - tzdata==2024.2
-      - uri-template==1.3.0
-      - uvicorn==0.32.1
-      - uvloop==0.21.0
-      - watchfiles==1.0.0
-      - wcwidth==0.2.13
-      - webcolors==24.11.1
-      - webencodings==0.5.1
-      - websockets==14.1
-      - werkzeug==3.1.3
-      - widgetsnbextension==4.0.13
-prefix: /home/zeus/miniconda3/envs/cloudspace
+    - lime