fix: fix minigpt-env wrong torch deps

This commit is contained in:
sharpbai 2023-04-22 02:12:10 +08:00
parent d93f715165
commit 0a781bf14a

View File

@ -3,14 +3,15 @@ channels:
- pytorch - pytorch
- defaults - defaults
- anaconda - anaconda
- nvidia
dependencies: dependencies:
- python=3.9 - python=3.9
- cudatoolkit - cudatoolkit
- pip - pip
- pytorch=1.12.1 - pytorch=1.13.1
- pytorch-mutex=1.0=cuda - pytorch-mutex=1.0=cuda
- torchaudio=0.12.1 - torchaudio=0.13.1
- torchvision=0.13.1 - torchvision=0.14.1
- pip: - pip:
- accelerate==0.16.0 - accelerate==0.16.0
- aiohttp==3.8.4 - aiohttp==3.8.4