소스 검색

Use official nvidia-ml-py instead of pynvml

Gaetan Lepage 10 달 전
부모
커밋
9ffd8116f0
1개의 변경된 파일1개의 추가작업 그리고 1개의 파일을 삭제
  1. 1 1
      setup.py

+ 1 - 1
setup.py

@@ -12,11 +12,11 @@ install_requires = [
   "Jinja2==3.1.4",
   "netifaces==0.11.0",
   "numpy==2.0.0",
+  "nvidia-ml-py==12.560.30",
   "pillow==10.4.0",
   "prometheus-client==0.20.0",
   "protobuf==5.27.1",
   "psutil==6.0.0",
-  "pynvml==11.5.3",
   "requests==2.32.3",
   "rich==13.7.1",
   "safetensors==0.4.3",