The code is tested on python 3.8, Pytorch 1.13
.
conda create -n FSMNet python=3.8
source activate FSMNet # or conda activate FSMNet
pip install torch==1.13.1+cu116 torchvision==0.14.1+cu116 torchaudio==0.13.1 --extra-index-url https://download.pytorch.org/whl/cu116
pip install einops h5py matplotlib scikit_image tensorboardX yacs pandas opencv-python timm ml_collections