Installation

The code was tested with Anaconda Python 3.9.12, CUDA 10.1, and PyTorch 1.10.1. After installing Anaconda:

[Optional but highly recommended] create a new conda environment.
```
conda create -n fullrot python=3.9.12
```
And activate the environment.
```
conda activate fullrot
```

Install PyTorch:

conda install pytorch=1.11.0 torchvision=0.12.0 cudatoolkit=11.3.1 -c pytorch

Clone this repo:

git clone https://github.com/anthonyweidai/FullRot_WRMix.git

Install the requirements

cd $FullRot_WRMix_ROOT
pip install -r requirements.txt