The code has been tested with Python 3.9.13 and PyTorch 1.13.0 with CUDA 11.7. conda create --name emos python=3.9 conda activate emos pip install torch==1.13.0 ...