Hugging Face's logo

Spaces:
soutrik
/
gradio_demo_CatDogClassifier
Runtime error

App Files Community
Soutrik commited on
Commit
854e05e
·
1 Parent(s): 1b0bd15

gpu vm

Browse files
Files changed (2) hide show
  1. poetry.lock +0 -0
  2. pyproject.toml +6 -5
poetry.lock CHANGED
The diff for this file is too large to render. See raw diff
 
pyproject.toml CHANGED
@@ -22,8 +22,9 @@ timm = "1.0.9"
22
  pandas = "^2.2.3"
23
  numpy = "^1.26.0"
24
  ruff = "^0.1.0"
25
- torch = {version = "^2.4.1+cpu", source = "pytorch_cpu"}
26
- torchvision = {version = "^0.19.1+cpu", source = "pytorch_cpu"}
 
27
  seaborn = "^0.13.2"
28
  pydantic = "^2.9.2"
29
  kaggle = "^1.6.17"
@@ -78,10 +79,10 @@ nvitop = "^1.3.2"
78
  pytest-asyncio = "^0.20.3"
79
 
80
  [[tool.poetry.source]]
81
- name = "pytorch_cpu"
82
- url = "https://download.pytorch.org/whl/cpu"
83
  priority = "explicit"
84
 
85
  [build-system]
86
  requires = ["poetry-core"]
87
- build-backend = "poetry.core.masonry.api"
 
22
  pandas = "^2.2.3"
23
  numpy = "^1.26.0"
24
  ruff = "^0.1.0"
25
+ torch = {version = "^2.4.1", source = "pytorch_cuda"}
26
+ torchvision = {version = "^0.19.1", source = "pytorch_cuda"}
27
+ torchaudio = {version = "^2.4.1", source = "pytorch_cuda"}
28
  seaborn = "^0.13.2"
29
  pydantic = "^2.9.2"
30
  kaggle = "^1.6.17"
 
79
  pytest-asyncio = "^0.20.3"
80
 
81
  [[tool.poetry.source]]
82
+ name = "pytorch_cuda"
83
+ url = "https://download.pytorch.org/whl/cu124"
84
  priority = "explicit"
85
 
86
  [build-system]
87
  requires = ["poetry-core"]
88
+ build-backend = "poetry.core.masonry.api"