malware-learning.def

Bootstrap: docker
From: pytorch/pytorch:1.7.0-cuda11.0-cudnn8-devel

%files
	requirements.txt /mnt/requirements.txt
	dgl-0.6-cp38-cp38-linux_x86_64.whl /mnt/dgl-0.6-cp38-cp38-linux_x86_64.whl

%post
    apt-get update && apt-get install -y git graphviz graphviz-dev && rm -rf /var/lib/apt/lists/*
	pip install -r /mnt/requirements.txt
	pip install /mnt/dgl-0.6-cp38-cp38-linux_x86_64.whl
	cd /mnt
	git clone --depth 1 https://github.com/androguard/androguard.git && cd androguard && python setup.py install