Yash Sachdeva commited on
Commit
bb0f9ac
1 Parent(s): 965dcf3
Files changed (1) hide show
  1. Dockerfile +1 -1
Dockerfile CHANGED
@@ -10,7 +10,7 @@ RUN pip install transformers
10
 
11
  # Install pytorch starts
12
 
13
- RUN conda install pytorch torchvision torchaudio cpuonly -c pytorch
14
 
15
  #Install pytorch ends
16
 
 
10
 
11
  # Install pytorch starts
12
 
13
+ RUN pip install torch
14
 
15
  #Install pytorch ends
16