This commit is contained in:
Xu Ma 2021-10-04 03:45:09 -04:00
parent 03eb2e48fd
commit 792665e763
2 changed files with 14 additions and 1 deletions

2
.gitignore vendored
View file

@ -5,7 +5,7 @@ __pycache__/
# C extensions # C extensions
*.so *.so
.idea
# Distribution / packaging # Distribution / packaging
.Python .Python
build/ build/

13
requirements.txt Normal file
View file

@ -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