INSTALL.md
July 20, 2022 ยท View on GitHub
Installation Instructions
- Clone this repo:
git clone https://github.com/Hxyou/MSCLIP
cd MSCLIP
- Create a conda virtual environment and activate it (optional):
conda create -n msclip python=3.7 -y
conda activate msclip
- Install
CUDA==10.1withcudnn7following the official installation instructions - Install
PyTorch==1.6.0andtorchvision==0.7.0withCUDA==10.1:
conda install pytorch==1.6.0 torchvision==0.7.0 cudatoolkit=10.1 -c pytorch
- Install
timm==0.3.2:
pip install timm==0.3.2
- Install other requirements:
pip install numpy scikit-learn Pillow tqdm yacs json ftfy regex transformers einops pickle tensorwatch