We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 60c61e3 commit 34fc656Copy full SHA for 34fc656
2 files changed
Dockerfile.tmpl
@@ -39,7 +39,6 @@ RUN uv pip install --system --force-reinstall --extra-index-url https://pypi.nvi
39
"nvidia-cuda-cupti-cu12==12.5.82" "nvidia-cuda-nvrtc-cu12==12.5.82" "nvidia-cuda-runtime-cu12==12.5.82" \
40
"nvidia-cufft-cu12==11.2.3.61" "nvidia-curand-cu12==10.3.6.82" "nvidia-cusparse-cu12==12.5.1.3" \
41
"nvidia-nvjitlink-cu12==12.5.82"
42
-
43
RUN uv pip install --system --force-reinstall "pynvjitlink-cu12==0.5.2"
44
45
# b/385145217 Latest Colab lacks mkl numpy, install it.
tests/test_torchtune.py
@@ -13,4 +13,4 @@ def test_help(self):
13
self.assertIn(
14
"Download a model from the Hugging Face Hub or Kaggle",
15
result.stdout
16
- )
+ )
0 commit comments