sangyun884 / HR-VITON

Official PyTorch implementation for the paper High-Resolution Virtual Try-On with Misalignment and Occlusion-Handled Conditions (ECCV 2022).
831 stars 171 forks source link

Package version - What is the correct? #38

Closed julianorinaldi closed 1 year ago

julianorinaldi commented 1 year ago

Could you specify what version each package have, because I'm trying with many versions and I don't can execute the project.

conda install pytorch torchvision torchaudio cudatoolkit=11.1 -c pytorch-lts -c nvidia

pip install opencv-python torchgeometry Pillow tqdm tensorboardX scikit-image scipy

pytorch= torchvision= torchaudio= cudatoolkit=11.1

opencv-python= torchgeometry= Pillow= tqdm= tensorboardX= scikit-image= scipy=

julianorinaldi commented 1 year ago

I resolved. Running on Google Colab:

1) conda install -y pytorch=1.8.1=py3.8_cuda11.1_cudnn8.0.50 torchaudio cudatoolkit=11.1.1 -c pytorch-lts -c nvidia -n base --no-pin 2) pip3 install -q --no-cache-dir opencv-python torchgeometry Pillow tqdm tensorboardX scikit-image scipy 3 pip3 install torchvision==0.9.1