search

YuanGongND / cav-mae

Code and Pretrained Models for ICLR 2023 Paper "Contrastive Audio-Visual Masked Autoencoder".
BSD 2-Clause "Simplified" License
223 stars 22 forks source link

installation #18

Closed chandlerbing65nm closed 11 months ago

chandlerbing65nm commented 11 months ago
  1. Create virtual environment using conda:
conda create --name venv python=3.8 -y
conda activate venv
  1. install compatible packages
pip install timm==0.4.5
pip install torch==1.12.0+cu116 torchvision==0.13.0+cu116 torchaudio==0.12.0 --extra-index-url https://download.pytorch.org/whl/cu116
pip install numpy==1.21.6
pip install scikit-learn