INSTALL.md
May 21, 2022 ยท View on GitHub
Installation Instructions
- Clone this repo:
git clone https://github.com/microsoft/UniCL
cd UniCL
- Create a conda virtual environment and activate it (optional):
conda create -n unicl python=3.7 -y
conda activate unicl
- Install
CUDA==10.1withcudnn7following the official installation instructions - Install
PyTorch==1.7.1andtorchvision==0.8.2withCUDA==10.1:
conda install pytorch==1.7.1 torchvision==0.8.2 cudatoolkit=10.1 -c pytorch
- Install
timm==0.3.2:
pip install timm==0.3.2
- 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" ./
- Install other requirements:
pip install opencv-python==4.4.0.46 termcolor==1.1.0 yacs==0.1.8