Skip to content

Commit 5feb0ce

Browse files
authored
Update Dockerfile
1 parent bec36b8 commit 5feb0ce

1 file changed

Lines changed: 1 addition & 5 deletions

File tree

images/scipy-ml-notebook/Dockerfile

Lines changed: 1 addition & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -54,25 +54,21 @@ RUN pip3 install --no-cache-dir --upgrade uv
5454

5555
RUN uv pip install --system \
5656
--extra-index-url https://pypi.nvidia.com \
57+
--extra-index-url https://download.pytorch.org/whl/cu126 \
5758
nvidia-cuda-nvcc-cu12 \
5859
nvidia-nccl-cu12 \
5960
cuda-python \
60-
--index-url "" \
6161
protobuf==$PROTOBUF_VERSION \
62-
# OpenCV
6362
opencv-contrib-python-headless \
6463
opencv-python \
65-
# Other Utils
6664
PyQt5 \
6765
pycocotools \
6866
pillow \
6967
scapy \
70-
--index-url https://download.pytorch.org/whl/cu126 \
7168
nvidia-cudnn-cu12==$CUDNN_VERSION \
7269
torch==$TORCH_VERSION \
7370
torchvision \
7471
torchaudio \
75-
--index-url "" \
7672
tensorflow==$TENSORFLOW_VERSION \
7773
tensorflow-datasets \
7874
tensorrt==$TENSORRT_VERSION \

0 commit comments

Comments
 (0)