Skip to content

Latest commit

 

History

History
28 lines (21 loc) · 833 Bytes

INSTALL.md

File metadata and controls

28 lines (21 loc) · 833 Bytes

Installation

  • Download the ImageNet dataset.

  • Install CUDA 11.3 with cuDNN 8 following the official installation guide of CUDA and cuDNN.

  • Setup conda environment:

# Create environment
conda create -n MFM python=3.8 -y
conda activate MFM

# Install requirements
conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch -y

# Install apex
git clone https://github.com/NVIDIA/apex
cd apex
pip install -v --disable-pip-version-check --no-cache-dir --global-option="--cpp_ext" --global-option="--cuda_ext" ./
cd ..

# Clone MFM
git clone https://github.com/Jiahao000/MFM
cd MFM

# Install other requirements
pip install -r requirements.txt