FcF-Inpainting / setup.sh
praeclarumjj3's picture
Print nvidia-smi
c2654f0
raw
history blame
201 Bytes
#!/bin/sh
nvidia-smi
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
python3 -c "import torch; print(torch.__version__)"