diff --git a/docker/base/fs/ins/install_python.sh b/docker/base/fs/ins/install_python.sh index 152dec77f..b17d17326 100644 --- a/docker/base/fs/ins/install_python.sh +++ b/docker/base/fs/ins/install_python.sh @@ -47,8 +47,8 @@ source /opt/venv/bin/activate pip install --no-cache-dir --upgrade pip ipython requests # Install some packages in specific variants pip install --no-cache-dir \ - torch==2.4.0 \ - torchvision==0.19.0 \ + torch==2.6.0 \ + torchvision==0.21.0 \ --index-url https://download.pytorch.org/whl/cpu echo "====================PYTHON UV ===================="