- Python 3.10
- PyTorch*
- Matplotlib
- Pymoo (https://pymoo.org/)
*Note: Runnable on PyTorch 1.11.0, and for MPS support you should see the official PyTorch documentation.
Could install with
# Install PyTorch with GPU support
conda install pytorch torchvision torchaudio pytorch-cuda=11.7 -c pytorch -c nvidia
conda install matplotlib
pip install pymoo
- PyNVML (for GPU monitoring in function
main_benchmarking
in main.py) - IPython (for interactive
sys.excepthook
)
Could install with
pip install pynvml
pip install ipython
Just run main.py with Python 3.10.
Configurable parameters are in main
function and problem_config/example.py.