Chore: Locust 패키지 추가 + torchvision 버전충돌 수정

2024-09-10 17:38:06 +09:00 · 2024-09-10 17:38:06 +09:00 · a5a4c49d62
commit a5a4c49d62
parent 75a12dd10c
1 changed files with 4 additions and 3 deletions
--- a/ai/environment.yml
+++ b/ai/environment.yml
@ -1,13 +1,13 @@
 name: worlabel_ai_env
 channels:
  - conda-forge
-  - nvidia
  - pytorch
+  - nvidia
  - defaults
 dependencies:
  - python=3.10.10
  - pytorch=2.3.1
-  - torchvision=0.16.1
+  - torchvision=0.18.1
  - torchaudio=2.3.1
  - pytorch-cuda=12.1
  - fastapi
@ -15,4 +15,5 @@ dependencies:
  - ultralytics
  - dill
  - boto3
-  - python-dotenv
+  - python-dotenv
+  - locust