Merge pull request #11 from shivammehta007/updating_packages

Updating packages
shivammehta25 · Jan 17, 2022 · 379e63d · 379e63d
2 parents 8be1690 + 7b3e019
commit 379e63d
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 7 deletions.
diff --git a/Dockerfile b/Dockerfile
@@ -1,4 +1,4 @@
-FROM pytorch/pytorch:1.9.0-cuda11.1-cudnn8-devel
+FROM nvcr.io/nvidia/pytorch:21.11-py3
 
 # We need a named user for ids to map correctly in VS Code
 RUN groupadd -r docker-user && useradd -r -m -s /bin/false -g docker-user docker-user

diff --git a/requirements.txt b/requirements.txt
@@ -1,10 +1,10 @@
-torch==1.9.0
-pytorch-lightning==1.4.9
-matplotlib==3.4.3
+torch==1.11.0a0+b6df043
 seaborn==0.11.2
-librosa==0.8.1
 nltk==3.6.5
+Unidecode==1.3.2                                                                                                                                                                                                                 
+notebook==6.4.4
+matplotlib==3.4.3
+pytorch-lightning==1.4.9
+librosa==0.8.1
 scipy==1.7.1
-Unidecode==1.3.2
 inflect==5.3.0
-notebook==6.4.4