We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent f764320 commit 09a3073Copy full SHA for 09a3073
1 file changed
pyproject.toml
@@ -21,7 +21,7 @@ dependencies = [
21
"torch>=2.11.0",
22
"torchvision>=0.26.0",
23
"transformers>=5.5.1",
24
- "nvidia-cuda-runtime<=12.0",
+ "nvidia-cuda-runtime==12.0.90",
25
]
26
27
authors = [
0 commit comments