Skip to content

GitLab

Switch branch/tag
Find file BlameHistoryPermalink
  • 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
README_Windows_CUDA_Acceleration_en_US.md 2.52 KB