pytorch_conda.yml

name: pytorch
channels:
    - conda-forge
    - fvcore
    - iopath
    - bottler
    - pytorch 
    - pytorch3d 

dependencies:
  
    #Install PyTorch 1.7.1
    - pytorch==1.7.1 
    - torchvision==0.8.2 
    - torchaudio==0.7.2 
    - cudatoolkit=11.0

    #Requirements for PyTorch3D. 
    - fvcore 
    - iopath
    - nvidiacub 

    #Install PyTorch3D. 
    - pytorch3d==0.4.0