pablovela5620 commited on
Commit
55eff55
1 Parent(s): 6a71966

Upload pyproject.toml with huggingface_hub

Browse files
Files changed (1) hide show
  1. 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"