KGM_conda_env_simple.yml

name: KGM
channels:
  - pytorch
  - nvidia/label/cuda-11.7.1
  - conda-forge
  - defaults
dependencies:
  - cuda-toolkit=11.7.1=0
  - cuda-tools=11.7.1=0
  - pip=23.3.1=py310h06a4308_0
  - python=3.10.13=h955ad1f_0
  - pytorch=2.0.1=py3.10_cuda11.7_cudnn8.5.0_0
  - torchvision=0.15.2=py310_cu117
  - wandb=0.16.1=pyhd8ed1ab_0
  - yaml=0.2.5=h7b6447c_0
  - pip:
      - numpy==1.23.5
      - plotly==5.20.0
      - tensorboard==2.15.1
      - pandas==2.1.1
      - matplotlib==3.8.0
      - scipy==1.11.4
      - thop
      - scikit-learn
      - pydantic
      - nltk
      - seaborn
      - IPython 
      - xlrd
      - rich
prefix: /home/xue/Repos/miniconda3/envs/taskGener