From 792665e763ec65be64c1b2b090027b56138a0c4b Mon Sep 17 00:00:00 2001 From: Xu Ma Date: Mon, 4 Oct 2021 03:45:09 -0400 Subject: [PATCH] update --- .gitignore | 2 +- requirements.txt | 13 +++++++++++++ 2 files changed, 14 insertions(+), 1 deletion(-) create mode 100644 requirements.txt diff --git a/.gitignore b/.gitignore index 99efc91..6440e67 100644 --- a/.gitignore +++ b/.gitignore @@ -5,7 +5,7 @@ __pycache__/ # C extensions *.so - +.idea # Distribution / packaging .Python build/ diff --git a/requirements.txt b/requirements.txt new file mode 100644 index 0000000..3c44c85 --- /dev/null +++ b/requirements.txt @@ -0,0 +1,13 @@ +cudatoolkit=10.2.89 +cycler=0.10.0 +einops=0.3.0 +h5py=3.2.1 +matplotlib=3.4.2 +numpy=1.20.2 +numpy-base=1.20.2 +pytorch=1.8.1 +pyyaml=5.4.1 +scikit-learn=0.24.2 +scipy=1.6.3 +torchvision=0.9.1 +tqdm=4.61.1