Skip to content

Commit 67297c4

Browse files
authored
[CI] Upgrade pytorch to 2.7.0, torchvision to 0.22.0, and vulkan sdk to 1.4.309 (#17887)
* upgrade pytorch and torchvision * upgrade vulkan
1 parent 0149fcd commit 67297c4

File tree

2 files changed

+12
-10
lines changed

2 files changed

+12
-10
lines changed

docker/install/ubuntu_install_onnx.sh

+8-8
Original file line numberDiff line numberDiff line change
@@ -44,12 +44,12 @@ if [ "$PYTHON_VERSION" == "3.9" ]; then
4444

4545
if [ "$DEVICE" == "cuda" ]; then
4646
pip3 install \
47-
torch==2.6.0 \
48-
torchvision==0.21.0
47+
torch==2.7.0 \
48+
torchvision==0.22.0
4949
else
5050
pip3 install \
51-
torch==2.6.0 \
52-
torchvision==0.21.0 \
51+
torch==2.7.0 \
52+
torchvision==0.22.0 \
5353
--extra-index-url https://download.pytorch.org/whl/cpu
5454
fi
5555
elif [ "$PYTHON_VERSION" == "3.11" ]; then
@@ -60,12 +60,12 @@ elif [ "$PYTHON_VERSION" == "3.11" ]; then
6060

6161
if [ "$DEVICE" == "cuda" ]; then
6262
pip3 install \
63-
torch==2.6.0 \
64-
torchvision==0.21.0
63+
torch==2.7.0 \
64+
torchvision==0.22.0
6565
else
6666
pip3 install \
67-
torch==2.6.0 \
68-
torchvision==0.21.0 \
67+
torch==2.7.0 \
68+
torchvision==0.22.0 \
6969
--extra-index-url https://download.pytorch.org/whl/cpu
7070
fi
7171
else

docker/install/ubuntu_install_vulkan.sh

+4-2
Original file line numberDiff line numberDiff line change
@@ -16,11 +16,13 @@
1616
# specific language governing permissions and limitations
1717
# under the License.
1818

19-
set -e
2019
set -u
2120
set -o pipefail
2221

22+
VULKAN_VERSION=1.4.309
23+
UBUNTU_VERSION=jammy
24+
2325
wget -qO - http://packages.lunarg.com/lunarg-signing-key-pub.asc | apt-key add -
24-
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
26+
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
2527
apt-get update
2628
apt-install-and-clear -y vulkan-sdk

0 commit comments

Comments
 (0)