- 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.1
withcudnn7
following the official installation instructions - Install
PyTorch==1.6.0
andtorchvision==0.7.0
withCUDA==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