pytorch_environment.yml

name: pytorch-1.5-cuda-10.1
channels:
    - conda-forge
    - defaults
    - pytorch
dependencies:
    - defaults::click=7.1.1
    - defaults::cudatoolkit=10.1
    - pytorch::pytorch=1.5.0=py3.7_cuda10.1.243_cudnn7.6.3_0
    - pytorch::torchvision=0.6.0=py37_cu101
    - conda-forge::graphviz=2.40.1
    - conda-forge::python-graphviz=0.13.2
    - pip
    - pip:
      - torchnet==0.0.4
      - alive-progress==2.4.1