We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent bec36b8 commit 5feb0ceCopy full SHA for 5feb0ce
1 file changed
images/scipy-ml-notebook/Dockerfile
@@ -54,25 +54,21 @@ RUN pip3 install --no-cache-dir --upgrade uv
54
55
RUN uv pip install --system \
56
--extra-index-url https://pypi.nvidia.com \
57
+ --extra-index-url https://download.pytorch.org/whl/cu126 \
58
nvidia-cuda-nvcc-cu12 \
59
nvidia-nccl-cu12 \
60
cuda-python \
- --index-url "" \
61
protobuf==$PROTOBUF_VERSION \
62
- # OpenCV
63
opencv-contrib-python-headless \
64
opencv-python \
65
- # Other Utils
66
PyQt5 \
67
pycocotools \
68
pillow \
69
scapy \
70
- --index-url https://download.pytorch.org/whl/cu126 \
71
nvidia-cudnn-cu12==$CUDNN_VERSION \
72
torch==$TORCH_VERSION \
73
torchvision \
74
torchaudio \
75
76
tensorflow==$TENSORFLOW_VERSION \
77
tensorflow-datasets \
78
tensorrt==$TENSORRT_VERSION \
0 commit comments