• myhloli's avatar
    build(deps): update torch and torchvision version requirements · 9a96362d
    myhloli authored
    - Specify torch==2.3.1 and torchvision==0.18.1 for Windows CUDA installation
    - Add torch and torchvision version constraints in setup.py:
      - torch>=2.2.2,<=2.3.1
      - torchvision>=0.17.2,<=0.18.1
    - Update installation instructions in both English and Chinese README files
    9a96362d
setup.py 3.69 KB