Hugging Face's logo

Spaces:
shi-labs
/
FcF-Inpainting
Runtime error

App Files Community
4
praeclarumjj3 commited on
Commit
c2654f0
•
1 Parent(s): 838e081

Print nvidia-smi

Browse files
Files changed (1) hide show
  1. setup.sh +3 -1
setup.sh CHANGED
@@ -1,2 +1,4 @@
1
  #!/bin/sh
2
- pip install torch==1.7.1+cu110 torchvision==0.8.2+cu110 torchaudio==0.7.2 -f https://download.pytorch.org/whl/torch_stable.html
 
 
 
1
  #!/bin/sh
2
+ nvidia-smi
3
+ pip3 install torch==1.7.1+cu110 torchvision==0.8.2+cu110 torchaudio==0.7.2 -f https://download.pytorch.org/whl/torch_stable.html
4
+ python3 -c "import torch; print(torch.__version__)"