diff --git a/docker/install/ubuntu_install_onnx.sh b/docker/install/ubuntu_install_onnx.sh index dc41c39d7c41..547f51fbb493 100755 --- a/docker/install/ubuntu_install_onnx.sh +++ b/docker/install/ubuntu_install_onnx.sh @@ -44,12 +44,12 @@ if [ "$PYTHON_VERSION" == "3.9" ]; then if [ "$DEVICE" == "cuda" ]; then pip3 install \ - torch==2.6.0 \ - torchvision==0.21.0 + torch==2.7.0 \ + torchvision==0.22.0 else pip3 install \ - torch==2.6.0 \ - torchvision==0.21.0 \ + torch==2.7.0 \ + torchvision==0.22.0 \ --extra-index-url https://download.pytorch.org/whl/cpu fi elif [ "$PYTHON_VERSION" == "3.11" ]; then @@ -60,12 +60,12 @@ elif [ "$PYTHON_VERSION" == "3.11" ]; then if [ "$DEVICE" == "cuda" ]; then pip3 install \ - torch==2.6.0 \ - torchvision==0.21.0 + torch==2.7.0 \ + torchvision==0.22.0 else pip3 install \ - torch==2.6.0 \ - torchvision==0.21.0 \ + torch==2.7.0 \ + torchvision==0.22.0 \ --extra-index-url https://download.pytorch.org/whl/cpu fi else diff --git a/docker/install/ubuntu_install_vulkan.sh b/docker/install/ubuntu_install_vulkan.sh index 78cd4143f83e..9bcc47bb1455 100755 --- a/docker/install/ubuntu_install_vulkan.sh +++ b/docker/install/ubuntu_install_vulkan.sh @@ -16,11 +16,13 @@ # specific language governing permissions and limitations # under the License. -set -e set -u set -o pipefail +VULKAN_VERSION=1.4.309 +UBUNTU_VERSION=jammy + wget -qO - http://packages.lunarg.com/lunarg-signing-key-pub.asc | apt-key add - -wget -qO /etc/apt/sources.list.d/lunarg-vulkan-1.2.135-xenial.list http://packages.lunarg.com/vulkan/1.2.135/lunarg-vulkan-1.2.135-xenial.list +wget -qO /etc/apt/sources.list.d/lunarg-vulkan-${VULKAN_VERSION}-${UBUNTU_VERSION}.list http://packages.lunarg.com/vulkan/${VULKAN_VERSION}/lunarg-vulkan-${VULKAN_VERSION}-${UBUNTU_VERSION}.list apt-get update apt-install-and-clear -y vulkan-sdk