torch_env_2_0_1.yml

name: torch_2.0.1
channels:
    - pytorch
    - nvidia
    - conda-forge
dependencies:
    - python=3.8
    - pytorch=2.0.1
    - torchvision
    - pytorch-cuda=11.7
    - ipykernel
    - pip
    - pip:
        - numpy
        - matplotlib
        - tableone
        - tqdm
        - pandas