Hugging Face's logo

Spaces:
yash009
/
textgeneration
Runtime error

App Files Community
Yash Sachdeva commited on
Commit
965dcf3
·
1 Parent(s): cc94792

pytorch

Browse files
Files changed (1) hide show
  1. Dockerfile +1 -2
Dockerfile CHANGED
@@ -10,8 +10,7 @@ RUN pip install transformers
10
 
11
  # Install pytorch starts
12
 
13
- RUN pip install https://download.pytorch.org/whl/cpu/torch-1.0.1-cp36-cp36m-win_amd64.whl
14
- RUN pip install torchvision
15
 
16
  #Install pytorch ends
17
 
 
10
 
11
  # Install pytorch starts
12
 
13
+ RUN conda install pytorch torchvision torchaudio cpuonly -c pytorch
 
14
 
15
  #Install pytorch ends
16