$ conda env create -f AXAI_no_cuda.yml
$ conda activate AXAI

Now you have everything you need to run the Jupiter notebook demo, AXAI_Tutorial.ipynb. We installed the non-cuda version Pytorch, so that everyone can run the code on their laptop. But if you have a machine with gpus then you can install the cuda version Pytorch. Our code runs even faster on GPUs.