environment.yml

name: nvresearch
channels:
  - pytorch
  - nvidia
  - nvidia/label/cuda-12.4.0
  - conda-forge
  - defaults
dependencies:
  - _libgcc_mutex=0.1=conda_forge
  - _openmp_mutex=4.5=2_gnu
  - anyio=4.2.0=py39h06a4308_0
  - argon2-cffi=21.3.0=pyhd3eb1b0_0
  - argon2-cffi-bindings=21.2.0=py39h7f8727e_0
  - asttokens=2.0.5=pyhd3eb1b0_0
  - async-lru=2.0.4=py39h06a4308_0
  - attrs=23.1.0=py39h06a4308_0
  - babel=2.11.0=py39h06a4308_0
  - backcall=0.2.0=pyhd3eb1b0_0
  - beautifulsoup4=4.12.3=py39h06a4308_0
  - blas=1.0=mkl
  - bleach=4.1.0=pyhd3eb1b0_0
  - bottleneck=1.3.7=py39ha9d4c09_0
  - brotli-python=1.0.9=py39h6a678d5_8
  - bzip2=1.0.8=h5eee18b_6
  - ca-certificates=2025.4.26=hbd8a1cb_0
  - certifi=2025.4.26=pyhd8ed1ab_0
  - cffi=1.16.0=py39h5eee18b_1
  - charset-normalizer=3.3.2=pyhd3eb1b0_0
  - comm=0.2.1=py39h06a4308_0
  - cuda=12.4.0=0
  - cuda-cccl=12.4.99=0
  - cuda-command-line-tools=12.4.0=0
  - cuda-compiler=12.4.0=0
  - cuda-cudart=12.4.99=0
  - cuda-cudart-dev=12.4.99=0
  - cuda-cudart-static=12.4.99=0
  - cuda-cuobjdump=12.4.99=0
  - cuda-cupti=12.4.99=0
  - cuda-cupti-static=12.4.99=0
  - cuda-cuxxfilt=12.4.99=0
  - cuda-demo-suite=12.4.99=0
  - cuda-documentation=12.4.99=0
  - cuda-driver-dev=12.4.99=0
  - cuda-gdb=12.4.99=0
  - cuda-libraries=12.4.0=0
  - cuda-libraries-dev=12.4.0=0
  - cuda-libraries-static=12.4.0=0
  - cuda-nsight=12.4.99=0
  - cuda-nsight-compute=12.4.0=0
  - cuda-nvcc=12.4.99=0
  - cuda-nvdisasm=12.4.99=0
  - cuda-nvml-dev=12.4.99=0
  - cuda-nvprof=12.4.99=0
  - cuda-nvprune=12.4.99=0
  - cuda-nvrtc=12.4.99=0
  - cuda-nvrtc-dev=12.4.99=0
  - cuda-nvrtc-static=12.4.99=0
  - cuda-nvtx=12.4.99=0
  - cuda-nvvp=12.4.99=0
  - cuda-opencl=12.4.99=0
  - cuda-opencl-dev=12.4.99=0
  - cuda-profiler-api=12.4.99=0
  - cuda-runtime=12.4.0=0
  - cuda-sanitizer-api=12.4.99=0
  - cuda-toolkit=12.4.0=0
  - cuda-tools=12.4.0=0
  - cuda-visual-tools=12.4.0=0
  - cyrus-sasl=2.1.28=h52b45da_1
  - dbus=1.13.18=hb2f20db_0
  - debugpy=1.6.7=py39h6a678d5_0
  - decorator=5.1.1=pyhd3eb1b0_0
  - defusedxml=0.7.1=pyhd3eb1b0_0
  - exceptiongroup=1.2.0=py39h06a4308_0
  - executing=0.8.3=pyhd3eb1b0_0
  - expat=2.6.2=h6a678d5_0
  - ffmpeg=4.3=hf484d3e_0
  - fontconfig=2.14.1=h55d465d_3
  - freetype=2.12.1=h4a9f257_0
  - gds-tools=1.9.0.20=0
  - glib=2.78.4=h6a678d5_0
  - glib-tools=2.78.4=h6a678d5_0
  - gmp=6.2.1=h295c915_3
  - gmpy2=2.1.2=py39heeb90bb_0
  - gnutls=3.6.15=he1e5248_0
  - gst-plugins-base=1.14.1=h6a678d5_1
  - gstreamer=1.14.1=h5eee18b_1
  - icu=73.1=h6a678d5_0
  - idna=3.7=py39h06a4308_0
  - importlib-metadata=7.0.1=py39h06a4308_0
  - importlib_metadata=7.0.1=hd3eb1b0_0
  - intel-openmp=2023.1.0=hdb19cb5_46306
  - ipykernel=6.28.0=py39h06a4308_0
  - ipython=8.15.0=py39h06a4308_0
  - ipywidgets=8.1.2=py39h06a4308_0
  - jedi=0.19.1=py39h06a4308_0
  - jpeg=9e=h5eee18b_3
  - json5=0.9.6=pyhd3eb1b0_0
  - jsonschema-specifications=2023.7.1=py39h06a4308_0
  - jupyter=1.0.0=py39h06a4308_9
  - jupyter-lsp=2.2.0=py39h06a4308_0
  - jupyter_client=8.6.0=py39h06a4308_0
  - jupyter_console=6.6.3=py39h06a4308_0
  - jupyter_core=5.7.2=py39h06a4308_0
  - jupyter_events=0.10.0=py39h06a4308_0
  - jupyter_server=2.14.1=py39h06a4308_0
  - jupyter_server_terminals=0.4.4=py39h06a4308_1
  - jupyterlab=4.0.11=py39h06a4308_0
  - jupyterlab_pygments=0.1.2=py_0
  - jupyterlab_server=2.25.1=py39h06a4308_0
  - jupyterlab_widgets=3.0.10=py39h06a4308_0
  - krb5=1.20.1=h143b758_1
  - lame=3.100=h7b6447c_0
  - lcms2=2.12=h3be6417_0
  - ld_impl_linux-64=2.38=h1181459_1
  - lerc=3.0=h295c915_0
  - libclang=14.0.6=default_hc6dbbc7_1
  - libclang13=14.0.6=default_he11475f_1
  - libcublas=12.4.2.65=0
  - libcublas-dev=12.4.2.65=0
  - libcublas-static=12.4.2.65=0
  - libcufft=11.2.0.44=0
  - libcufft-dev=11.2.0.44=0
  - libcufft-static=11.2.0.44=0
  - libcufile=1.9.0.20=0
  - libcufile-dev=1.9.0.20=0
  - libcufile-static=1.9.0.20=0
  - libcups=2.4.2=h2d74bed_1
  - libcurand=10.3.5.119=0
  - libcurand-dev=10.3.5.119=0
  - libcurand-static=10.3.5.119=0
  - libcusolver=11.6.0.99=0
  - libcusolver-dev=11.6.0.99=0
  - libcusolver-static=11.6.0.99=0
  - libcusparse=12.3.0.142=0
  - libcusparse-dev=12.3.0.142=0
  - libcusparse-static=12.3.0.142=0
  - libdeflate=1.17=h5eee18b_1
  - libedit=3.1.20230828=h5eee18b_0
  - libffi=3.4.4=h6a678d5_1
  - libgcc=12.4.0=h767d61c_2
  - libgcc-ng=12.4.0=h69a702a_2
  - libglib=2.78.4=hdc74915_0
  - libgomp=12.4.0=h767d61c_2
  - libiconv=1.16=h5eee18b_3
  - libidn2=2.3.4=h5eee18b_0
  - libjpeg-turbo=2.0.0=h9bf148f_0
  - libllvm14=14.0.6=hdb19cb5_3
  - libnpp=12.2.5.2=0
  - libnpp-dev=12.2.5.2=0
  - libnpp-static=12.2.5.2=0
  - libnvfatbin=12.4.99=0
  - libnvfatbin-dev=12.4.99=0
  - libnvjitlink=12.4.99=0
  - libnvjitlink-dev=12.4.99=0
  - libnvjpeg=12.3.1.89=0
  - libnvjpeg-dev=12.3.1.89=0
  - libnvjpeg-static=12.3.1.89=0
  - libpng=1.6.39=h5eee18b_0
  - libpq=12.17=hdbd6064_0
  - libsodium=1.0.18=h7b6447c_0
  - libstdcxx=12.4.0=h8f9b012_2
  - libstdcxx-ng=12.4.0=h4852527_2
  - libtasn1=4.19.0=h5eee18b_0
  - libtiff=4.5.1=h6a678d5_0
  - libunistring=0.9.10=h27cfd23_0
  - libuuid=1.41.5=h5eee18b_0
  - libwebp-base=1.3.2=h5eee18b_0
  - libxcb=1.15=h7f8727e_0
  - libxkbcommon=1.0.1=h097e994_2
  - libxml2=2.13.1=hfdd30dd_2
  - llvm-openmp=14.0.6=h9e868ea_0
  - lz4-c=1.9.4=h6a678d5_1
  - markupsafe=2.1.3=py39h5eee18b_0
  - matplotlib-inline=0.1.6=py39h06a4308_0
  - mistune=2.0.4=py39h06a4308_0
  - mkl=2023.1.0=h213fc3f_46344
  - mkl-service=2.4.0=py39h5eee18b_1
  - mkl_fft=1.3.8=py39h5eee18b_0
  - mkl_random=1.2.4=py39hdb19cb5_0
  - mpc=1.1.0=h10f8cd9_1
  - mpfr=4.0.2=hb69a4c5_1
  - mpmath=1.3.0=py39h06a4308_0
  - mysql=5.7.24=h721c034_2
  - nbclient=0.8.0=py39h06a4308_0
  - nbconvert=7.10.0=py39h06a4308_0
  - nbformat=5.9.2=py39h06a4308_0
  - ncurses=6.5=he02047a_1
  - nest-asyncio=1.6.0=py39h06a4308_0
  - nettle=3.7.3=hbbd107a_1
  - networkx=3.2.1=py39h06a4308_0
  - notebook=7.0.8=py39h06a4308_2
  - notebook-shim=0.2.3=py39h06a4308_0
  - nsight-compute=2024.1.0.13=0
  - numexpr=2.8.7=py39h85018f9_0
  - numpy=1.26.4=py39h5f9d8c6_0
  - numpy-base=1.26.4=py39hb5e798b_0
  - openh264=2.1.1=h4ff587b_0
  - openjpeg=2.5.2=he7f1fd0_0
  - openssl=3.3.1=h4bc722e_2
  - overrides=7.4.0=py39h06a4308_0
  - packaging=24.1=py39h06a4308_0
  - pandas=1.5.3=py39h417a72b_0
  - pandocfilters=1.5.0=pyhd3eb1b0_0
  - parso=0.8.3=pyhd3eb1b0_0
  - pcre2=10.42=hebb0a14_1
  - pexpect=4.8.0=pyhd3eb1b0_3
  - pickleshare=0.7.5=pyhd3eb1b0_1003
  - pillow=10.4.0=py39h5eee18b_0
  - pip=24.0=py39h06a4308_0
  - platformdirs=3.10.0=py39h06a4308_0
  - ply=3.11=py39h06a4308_0
  - prompt-toolkit=3.0.43=py39h06a4308_0
  - prompt_toolkit=3.0.43=hd3eb1b0_0
  - psutil=5.9.0=py39h5eee18b_0
  - ptyprocess=0.7.0=pyhd3eb1b0_2
  - pure_eval=0.2.2=pyhd3eb1b0_0
  - pycparser=2.21=pyhd3eb1b0_0
  - pygments=2.15.1=py39h06a4308_1
  - pyqt=5.15.10=py39h6a678d5_0
  - pyqt5-sip=12.13.0=py39h5eee18b_0
  - pysocks=1.7.1=py39h06a4308_0
  - python=3.9.19=h955ad1f_1
  - python-dateutil=2.9.0post0=py39h06a4308_2
  - python-fastjsonschema=2.16.2=py39h06a4308_0
  - python-json-logger=2.0.7=py39h06a4308_0
  - python-tzdata=2023.3=pyhd3eb1b0_0
  - pytorch-cuda=12.4=hc786d27_6
  - pytorch-mutex=1.0=cuda
  - pytz=2024.1=py39h06a4308_0
  - pyyaml=6.0.1=py39h5eee18b_0
  - pyzmq=25.1.2=py39h6a678d5_0
  - qt-main=5.15.2=h53bd1ea_10
  - qtconsole=5.5.1=py39h06a4308_0
  - qtpy=2.4.1=py39h06a4308_0
  - readline=8.2=h5eee18b_0
  - referencing=0.30.2=py39h06a4308_0
  - requests=2.32.3=py39h06a4308_0
  - rfc3339-validator=0.1.4=py39h06a4308_0
  - rfc3986-validator=0.1.1=py39h06a4308_0
  - rpds-py=0.10.6=py39hb02cf49_0
  - send2trash=1.8.2=py39h06a4308_0
  - setuptools=72.1.0=py39h06a4308_0
  - sip=6.7.12=py39h6a678d5_0
  - six=1.16.0=pyhd3eb1b0_1
  - sniffio=1.3.0=py39h06a4308_0
  - soupsieve=2.5=py39h06a4308_0
  - sqlite=3.45.3=h5eee18b_0
  - stack_data=0.2.0=pyhd3eb1b0_0
  - tbb=2021.8.0=hdb19cb5_0
  - terminado=0.17.1=py39h06a4308_0
  - tinycss2=1.2.1=py39h06a4308_0
  - tk=8.6.14=h39e8969_0
  - tomli=2.0.1=py39h06a4308_0
  - tornado=6.4.1=py39h5eee18b_0
  - traitlets=5.14.3=py39h06a4308_0
  - tzdata=2024a=h04d1e81_0
  - urllib3=2.2.2=py39h06a4308_0
  - wcwidth=0.2.5=pyhd3eb1b0_0
  - webencodings=0.5.1=py39h06a4308_1
  - websocket-client=1.8.0=py39h06a4308_0
  - wheel=0.43.0=py39h06a4308_0
  - widgetsnbextension=4.0.10=py39h06a4308_0
  - xz=5.4.6=h5eee18b_1
  - yaml=0.2.5=h7b6447c_0
  - zeromq=4.3.5=h6a678d5_0
  - zipp=3.17.0=py39h06a4308_0
  - zlib=1.2.13=h5eee18b_1
  - zstd=1.5.5=hc292b87_2
  - pip:
      - absl-py==2.1.0
      - accelerate==0.33.0
      - adjusttext==1.3.0
      - aiohappyeyeballs==2.3.4
      - aiohttp==3.10.1
      - aiosignal==1.3.1
      - airportsdata==20250224
      - altair==5.4.1
      - ampselgemm==0.0.0
      - annotated-types==0.7.0
      - anykeystore==0.2
      - apex==0.1
      - astor==0.8.1
      - async-timeout==4.0.3
      - balanced-kmeans==0.1.0
      - basicgemm==0.0.0
      - bitsandbytes==0.43.3
      - blake3==1.0.4
      - blinker==1.9.0
      - cachetools==5.5.0
      - chardet==5.2.0
      - click==8.1.7
      - cloudpickle==3.1.1
      - colorama==0.4.6
      - compressed-tensors==0.9.2
      - contexttimer==0.3.3
      - contourpy==1.2.1
      - cryptacular==1.6.2
      - cuda-python==12.6.0
      - cupy-cuda12x==13.4.1
      - cycler==0.12.1
      - dataproperty==1.1.0
      - datasets==2.20.0
      - depyf==0.18.0
      - dill==0.3.8
      - diskcache==5.6.3
      - distro==1.9.0
      - dnspython==2.7.0
      - einops==0.8.0
      - email-validator==2.2.0
      - evaluate==0.4.3
      - fastapi==0.115.12
      - fastapi-cli==0.0.7
      - fastrlock==0.8.3
      - filelock==3.18.0
      - flash-attn==2.6.3
      - fonttools==4.53.1
      - frozenlist==1.4.1
      - fsspec==2024.5.0
      - fused-dense-lib==0.0.0
      - fuzzywuzzy==0.18.0
      - gguf==0.10.0
      - gitdb==4.0.11
      - gitpython==3.1.43
      - greenlet==3.0.3
      - grouped-gemm-v1==0.0.0
      - h11==0.14.0
      - httpcore==1.0.7
      - httptools==0.6.4
      - httpx==0.28.1
      - huggingface-hub==0.28.1
      - hupper==1.12.1
      - immutabledict==4.2.1
      - importlib-resources==6.4.0
      - iniconfig==2.0.0
      - interegular==0.3.3
      - jieba==0.42.1
      - jinja2==3.1.6
      - jiter==0.9.0
      - joblib==1.4.2
      - jsonlines==4.0.0
      - jsonschema==4.23.0
      - k-means-constrained==0.7.3
      - kiwisolver==1.4.5
      - lark==1.2.2
      - lightning-utilities==0.11.9
      - llguidance==0.7.11
      - llvmlite==0.43.0
      - lm-eval==0.4.8
      - lm-format-enforcer==0.10.11
      - lxml==5.3.0
      - markdown-it-py==3.0.0
      - matplotlib==3.9.0
      - mbstrdecoder==1.1.4
      - mdurl==0.1.2
      - mistral-common==1.5.4
      - more-itertools==10.6.0
      - msgpack==1.1.0
      - msgspec==0.19.0
      - multidict==6.0.5
      - multiprocess==0.70.16
      - narwhals==1.13.5
      - ninja==1.11.1.4
      - nltk==3.9.1
      - numba==0.60.0
      - nvidia-cublas-cu12==12.4.5.8
      - nvidia-cuda-cupti-cu12==12.4.127
      - nvidia-cuda-nvrtc-cu12==12.4.127
      - nvidia-cuda-runtime-cu12==12.4.127
      - nvidia-cudnn-cu12==9.1.0.70
      - nvidia-cufft-cu12==11.2.1.3
      - nvidia-curand-cu12==10.3.5.147
      - nvidia-cusolver-cu12==11.6.1.9
      - nvidia-cusparse-cu12==12.3.1.170
      - nvidia-cusparselt-cu12==0.6.2
      - nvidia-cutlass==3.5.0.0
      - nvidia-nccl-cu12==2.21.5
      - nvidia-nvjitlink-cu12==12.4.127
      - nvidia-nvtx-cu12==12.4.127
      - oauthlib==3.2.2
      - openai==1.70.0
      - opencv-python-headless==4.11.0.86
      - ortools==9.11.4210
      - outlines==0.1.11
      - outlines-core==0.1.26
      - partial-json-parser==0.2.1.1.post5
      - pastedeploy==3.1.0
      - pathvalidate==3.2.3
      - pbkdf2==1.3
      - peft==0.14.0
      - plaster==1.1.2
      - plaster-pastedeploy==1.0.1
      - pluggy==1.5.0
      - portalocker==3.1.1
      - prometheus-client==0.21.1
      - prometheus-fastapi-instrumentator==7.1.0
      - protobuf==5.26.1
      - py-cpuinfo==9.0.0
      - pyarrow==17.0.0
      - pyarrow-hotfix==0.6
      - pybind11==2.13.6
      - pycountry==24.6.1
      - pydantic==2.11.1
      - pydantic-core==2.33.0
      - pydeck==0.9.1
      - pydot==3.0.1
      - pyparsing==3.1.2
      - pyramid==1.5.8
      - pyramid-mailer==0.15.1
      - pytablewriter==1.2.1
      - pytest==8.3.3
      - python-dotenv==1.1.0
      - python-multipart==0.0.20
      - python3-openid==3.2.0
      - ray==2.44.1
      - regex==2024.7.24
      - repoze-lru==0.7
      - repoze-sendmail==4.4.1
      - requests-oauthlib==2.0.0
      - rich==13.9.4
      - rich-toolkit==0.14.1
      - rouge==1.0.1
      - rouge-score==0.1.2
      - sacrebleu==2.5.1
      - safetensors==0.4.4
      - scikit-learn==1.5.2
      - scipy==1.13.1
      - seaborn==0.13.2
      - sentencepiece==0.2.0
      - shellingham==1.5.4
      - smmap==5.0.1
      - sqlalchemy==2.0.32
      - sqlitedict==2.1.0
      - starlette==0.46.1
      - streamlit==1.40.1
      - sympy==1.13.1
      - tabledata==1.3.4
      - tabulate==0.9.0
      - tcolorpy==0.1.7
      - tenacity==9.0.0
      - threadpoolctl==3.5.0
      - tiktoken==0.9.0
      - tokenizers==0.21.1
      - toml==0.10.2
      - torch==2.6.0
      - torchaudio==2.6.0
      - torchmetrics==1.6.0
      - torchvision==0.21.0
      - tqdm==4.66.5
      - tqdm-multiprocess==0.0.11
      - transaction==4.0
      - transformers==4.50.3
      - translationstring==1.4
      - treelib==1.7.0
      - triton==3.2.0
      - triton-nightly==3.0.0.post20240716052845
      - typepy==1.3.4
      - typer==0.15.2
      - typing-extensions==4.13.0
      - typing-inspection==0.4.0
      - uvicorn==0.34.0
      - uvloop==0.21.0
      - velruse==1.1.1
      - venusian==3.1.0
      - vllm==0.8.2
      - watchdog==6.0.0
      - watchfiles==1.0.4
      - webob==1.8.7
      - websockets==15.0.1
      - word2number==1.1
      - wtforms==3.1.2
      - wtforms-recaptcha==0.3.2
      - xformers==0.0.29.post2
      - xgrammar==0.1.16
      - xxhash==3.4.1
      - yarl==1.9.4
      - zope-deprecation==5.0
      - zope-interface==7.0.1
      - zope-sqlalchemy==3.1
      - zstandard==0.23.0