update
This commit is contained in:
parent
03eb2e48fd
commit
792665e763
2
.gitignore
vendored
2
.gitignore
vendored
|
@ -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
13
requirements.txt
Normal 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
|
Loading…
Reference in a new issue