Spaces:
Running
on
Zero
Running
on
Zero
pablovela5620
commited on
Upload pyproject.toml with huggingface_hub
Browse files- pyproject.toml +2 -0
pyproject.toml
CHANGED
@@ -45,6 +45,8 @@ torchvision = { version = "0.18.1", channel = "pytorch" }
|
|
45 |
cuda = { version = "*", channel = "nvidia/label/cuda-11.8.0" }
|
46 |
pytorch-cuda = { version = "11.8.*", channel = "pytorch" }
|
47 |
cuda-nvcc = "11.8.*"
|
|
|
|
|
48 |
beartype = ">=0.18.5,<0.19"
|
49 |
rerun-sdk = ">=0.18.2,<0.19"
|
50 |
jaxtyping = ">=0.2.34,<0.3"
|
|
|
45 |
cuda = { version = "*", channel = "nvidia/label/cuda-11.8.0" }
|
46 |
pytorch-cuda = { version = "11.8.*", channel = "pytorch" }
|
47 |
cuda-nvcc = "11.8.*"
|
48 |
+
cuda-cccl = "11.8.*"
|
49 |
+
cuda-opencl = "12.1.*"
|
50 |
beartype = ">=0.18.5,<0.19"
|
51 |
rerun-sdk = ">=0.18.2,<0.19"
|
52 |
jaxtyping = ">=0.2.34,<0.3"
|