Chore: environment.yml 패키지 수정

2024-09-05 17:07:25 +09:00 · 2024-09-05 17:07:25 +09:00 · 9abf99b992
commit 9abf99b992
parent 1196476286
1 changed files with 7 additions and 4 deletions
--- a/ai/environment.yml
+++ b/ai/environment.yml
@ -6,10 +6,13 @@ channels:
  - defaults
 dependencies:
  - python=3.10.10
-  - pytorch
-  - torchvision
-  - torchaudio
+  - pytorch=2.3.1
+  - torchvision=0.16.1
+  - torchaudio=2.3.1
  - pytorch-cuda=12.1
  - fastapi
  - uvicorn
-  - ultralytics
+  - ultralytics
+  - dill
+  - boto3
+  - python-dotenv