diff --git a/environment.yml b/environment.yml index b596b15..7709276 100644 --- a/environment.yml +++ b/environment.yml @@ -1,5 +1,7 @@ name: dtsemnet channels: + - pytorch + - nvidia - conda-forge - defaults dependencies: @@ -7,26 +9,106 @@ dependencies: - _openmp_mutex=5.1=1_gnu - asttokens=2.4.1=pyhd8ed1ab_0 - backcall=0.2.0=pyh9f0ad1d_0 + - blas=2.16=mkl + - brotli-python=1.0.9=py39h5a03fae_7 + - bzip2=1.0.8=h7f98852_4 - ca-certificates=2024.2.2=hbcca054_0 + - certifi=2024.2.2=pyhd8ed1ab_0 + - charset-normalizer=3.3.2=pyhd8ed1ab_0 - comm=0.2.2=pyhd8ed1ab_0 + - cuda=11.6.1=0 + - cuda-cccl=11.6.55=hf6102b2_0 + - cuda-command-line-tools=11.6.2=0 + - cuda-compiler=11.6.2=0 + - cuda-cudart=11.6.55=he381448_0 + - cuda-cudart-dev=11.6.55=h42ad0f4_0 + - cuda-cuobjdump=11.6.124=h2eeebcb_0 + - cuda-cupti=11.6.124=h86345e5_0 + - cuda-cuxxfilt=11.6.124=hecbf4f6_0 + - cuda-driver-dev=11.6.55=0 + - cuda-gdb=12.4.127=0 + - cuda-libraries=11.6.1=0 + - cuda-libraries-dev=11.6.1=0 + - cuda-memcheck=11.8.86=0 + - cuda-nsight=12.4.127=0 + - cuda-nsight-compute=12.4.1=0 + - cuda-nvcc=11.6.124=hbba6d2d_0 + - cuda-nvdisasm=12.4.127=0 + - cuda-nvml-dev=11.6.55=haa9ef22_0 + - cuda-nvprof=12.4.127=0 + - cuda-nvprune=11.6.124=he22ec0a_0 + - cuda-nvrtc=11.6.124=h020bade_0 + - cuda-nvrtc-dev=11.6.124=h249d397_0 + - cuda-nvtx=11.6.124=h0630a44_0 + - cuda-nvvp=12.4.127=0 + - cuda-runtime=11.6.1=0 + - cuda-samples=11.6.101=h8efea70_0 + - cuda-sanitizer-api=12.4.127=0 + - cuda-toolkit=11.6.1=0 + - cuda-tools=11.6.1=0 + - cuda-visual-tools=11.6.1=0 - debugpy=1.6.7=py39h6a678d5_0 - decorator=5.1.1=pyhd8ed1ab_0 - entrypoints=0.4=pyhd8ed1ab_0 - executing=2.0.1=pyhd8ed1ab_0 + - ffmpeg=4.3=hf484d3e_0 + - freetype=2.10.4=h0708190_1 + - gds-tools=1.9.1.3=0 + - gmp=6.2.1=h58526e2_0 + - gnutls=3.6.13=h85f3911_1 + - intel-openmp=2023.1.0=hdb19cb5_46306 - ipykernel=6.29.3=pyhd33586a_0 - ipython=8.12.0=pyh41d4057_0 + - jbig=2.1=h7f98852_2003 - jedi=0.19.1=pyhd8ed1ab_0 + - jpeg=9e=h166bdaf_1 - jupyter_client=7.3.4=pyhd8ed1ab_0 - jupyter_core=5.7.2=py39hf3d152e_0 + - lame=3.100=h7f98852_1001 + - lcms2=2.12=hddcbb42_0 - ld_impl_linux-64=2.38=h1181459_1 + - lerc=2.2.1=h9c3ff4c_0 + - libblas=3.8.0=16_mkl + - libcblas=3.8.0=16_mkl + - libcublas=11.9.2.110=h5e84587_0 + - libcublas-dev=11.9.2.110=h5c901ab_0 + - libcufft=10.7.1.112=hf425ae0_0 + - libcufft-dev=10.7.1.112=ha5ce4c0_0 + - libcufile=1.9.1.3=0 + - libcufile-dev=1.9.1.3=0 + - libcurand=10.3.5.147=0 + - libcurand-dev=10.3.5.147=0 + - libcusolver=11.3.4.124=h33c3c4e_0 + - libcusparse=11.7.2.124=h7538f96_0 + - libcusparse-dev=11.7.2.124=hbbe9722_0 + - libdeflate=1.7=h7f98852_5 - libffi=3.4.4=h6a678d5_0 - libgcc-ng=11.2.0=h1234567_1 + - libgfortran-ng=7.5.0=h14aa051_20 + - libgfortran4=7.5.0=h14aa051_20 - libgomp=11.2.0=h1234567_1 + - libiconv=1.17=h166bdaf_0 + - liblapack=3.8.0=16_mkl + - liblapacke=3.8.0=16_mkl + - libnpp=11.6.3.124=hd2722f0_0 + - libnpp-dev=11.6.3.124=h3c42840_0 + - libnvjpeg=11.6.2.124=hd473ad6_0 + - libnvjpeg-dev=11.6.2.124=hb5906b9_0 + - libpng=1.6.37=h21135ba_2 - libsodium=1.0.18=h36c2ea0_1 - libstdcxx-ng=11.2.0=h1234567_1 + - libtiff=4.3.0=hf544144_1 + - libwebp-base=1.2.2=h7f98852_1 + - lz4-c=1.9.3=h9c3ff4c_1 - matplotlib-inline=0.1.6=pyhd8ed1ab_0 + - mkl=2020.2=256 - ncurses=6.4=h6a678d5_0 - nest-asyncio=1.6.0=pyhd8ed1ab_0 + - nettle=3.6=he412f7d_0 + - nsight-compute=2024.1.1.4=0 + - olefile=0.47=pyhd8ed1ab_0 + - openh264=2.1.1=h780b84a_0 + - openjpeg=2.4.0=hb52868f_1 - openssl=3.0.13=h7f8727e_0 - parso=0.8.4=pyhd8ed1ab_0 - pexpect=4.9.0=pyhd8ed1ab_0 @@ -39,39 +121,55 @@ dependencies: - ptyprocess=0.7.0=pyhd3deb0d_0 - pure_eval=0.2.2=pyhd8ed1ab_0 - pygments=2.17.2=pyhd8ed1ab_0 + - pysocks=1.7.1=pyha2e5f31_6 - python=3.9.18=h955ad1f_0 - python_abi=3.9=2_cp39 + - pytorch=1.13.1=py3.9_cuda11.6_cudnn8.3.2_0 + - pytorch-cuda=11.6=h867d48c_1 + - pytorch-mutex=1.0=cuda - pyzmq=25.1.2=py39h6a678d5_0 - readline=8.2=h5eee18b_0 - six=1.16.0=pyh6c4a22f_0 - sqlite=3.41.2=h5eee18b_0 - stack_data=0.6.2=pyhd8ed1ab_0 - tk=8.6.12=h1ccaba5_0 + - torchaudio=0.13.1=py39_cu116 + - torchvision=0.14.1=py39_cu116 - tornado=6.1=py39hb9d737c_3 - traitlets=5.14.2=pyhd8ed1ab_0 - typing_extensions=4.11.0=pyha770c72_0 - tzdata=2024a=h04d1e81_0 + - urllib3=2.2.1=pyhd8ed1ab_0 - wcwidth=0.2.13=pyhd8ed1ab_0 - wheel=0.41.2=py39h06a4308_0 - xz=5.4.5=h5eee18b_0 - zeromq=4.3.5=h6a678d5_0 - zlib=1.2.13=h5eee18b_0 + - zstd=1.5.0=ha95c52a_0 - pip: - absl-py==2.1.0 + - aiohttp==3.9.5 + - aiosignal==1.3.1 - astunparse==1.6.3 + - async-timeout==4.0.3 + - attrs==23.2.0 - cachetools==5.3.3 - - certifi==2024.2.2 - - charset-normalizer==3.3.2 - cycler==0.12.1 - cython==3.0.8 + - datasets==2.20.0 + - dill==0.3.8 + - filelock==3.15.1 - flatbuffers==23.5.26 - fonttools==4.49.0 + - frozenlist==1.4.1 + - fsspec==2024.5.0 - gast==0.5.4 - google-auth==2.28.1 - google-auth-oauthlib==1.2.0 - google-pasta==0.2.0 - grpcio==1.62.0 - h5py==3.10.0 + - huggingface-hub==0.23.4 - idna==3.6 - importlib-metadata==7.0.1 - joblib==1.3.2 @@ -85,26 +183,35 @@ dependencies: - matplotlib==3.5.1 - mdurl==0.1.2 - ml-dtypes==0.2.0 + - multidict==6.0.5 + - multiprocess==0.70.16 - numba==0.59.0 - numpy==1.26.4 - oauthlib==3.2.2 - opt-einsum==3.3.0 - packaging==23.2 - pandas==1.5.0 + - patsy==0.5.6 - pillow==10.2.0 - protobuf==4.25.3 + - pyarrow==16.1.0 + - pyarrow-hotfix==0.6 - pyasn1==0.5.1 - pyasn1-modules==0.3.0 - pyparsing==3.1.1 - python-dateutil==2.8.2 - pytz==2024.1 - - requests==2.31.0 + - pyyaml==6.0.1 + - requests==2.32.3 - requests-oauthlib==1.3.1 - rich==13.7.0 - rsa==4.9 - scikit-learn==1.4.1.post1 + - scikit-posthocs==0.9.0 - scipy==1.12.0 + - seaborn==0.13.2 - setuptools==59.8.0 + - statsmodels==0.14.2 - tensorboard==2.15.2 - tensorboard-data-server==0.7.2 - tensorflow==2.15.0.post1 @@ -112,12 +219,11 @@ dependencies: - tensorflow-io-gcs-filesystem==0.36.0 - termcolor==2.4.0 - threadpoolctl==3.3.0 - - torch==1.13.1+cu116 - - torchaudio==0.13.1+cu116 - - torchvision==0.14.1+cu116 + - tqdm==4.66.4 - typing-extensions==4.10.0 - - urllib3==2.2.1 - werkzeug==3.0.1 - wrapt==1.14.1 + - xxhash==3.4.1 + - yarl==1.9.4 - zipp==3.17.0