From 0a781bf14aff503d810b731e7ee2e63a1a62a035 Mon Sep 17 00:00:00 2001 From: sharpbai Date: Sat, 22 Apr 2023 02:12:10 +0800 Subject: [PATCH] fix: fix minigpt-env wrong torch deps --- environment.yml | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/environment.yml b/environment.yml index d5cfcf8..186069e 100644 --- a/environment.yml +++ b/environment.yml @@ -3,14 +3,15 @@ channels: - pytorch - defaults - anaconda + - nvidia dependencies: - python=3.9 - cudatoolkit - pip - - pytorch=1.12.1 + - pytorch=1.13.1 - pytorch-mutex=1.0=cuda - - torchaudio=0.12.1 - - torchvision=0.13.1 + - torchaudio=0.13.1 + - torchvision=0.14.1 - pip: - accelerate==0.16.0 - aiohttp==3.8.4