This commit is contained in:
sharpbai 2023-09-20 19:19:59 +08:00 committed by GitHub
commit 9c689a6da7
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

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