File tree 2 files changed +12
-10
lines changed
2 files changed +12
-10
lines changed Original file line number Diff line number Diff line change @@ -44,12 +44,12 @@ if [ "$PYTHON_VERSION" == "3.9" ]; then
44
44
45
45
if [ " $DEVICE " == " cuda" ]; then
46
46
pip3 install \
47
- torch==2.6 .0 \
48
- torchvision==0.21 .0
47
+ torch==2.7 .0 \
48
+ torchvision==0.22 .0
49
49
else
50
50
pip3 install \
51
- torch==2.6 .0 \
52
- torchvision==0.21 .0 \
51
+ torch==2.7 .0 \
52
+ torchvision==0.22 .0 \
53
53
--extra-index-url https://download.pytorch.org/whl/cpu
54
54
fi
55
55
elif [ " $PYTHON_VERSION " == " 3.11" ]; then
@@ -60,12 +60,12 @@ elif [ "$PYTHON_VERSION" == "3.11" ]; then
60
60
61
61
if [ " $DEVICE " == " cuda" ]; then
62
62
pip3 install \
63
- torch==2.6 .0 \
64
- torchvision==0.21 .0
63
+ torch==2.7 .0 \
64
+ torchvision==0.22 .0
65
65
else
66
66
pip3 install \
67
- torch==2.6 .0 \
68
- torchvision==0.21 .0 \
67
+ torch==2.7 .0 \
68
+ torchvision==0.22 .0 \
69
69
--extra-index-url https://download.pytorch.org/whl/cpu
70
70
fi
71
71
else
Original file line number Diff line number Diff line change 16
16
# specific language governing permissions and limitations
17
17
# under the License.
18
18
19
- set -e
20
19
set -u
21
20
set -o pipefail
22
21
22
+ VULKAN_VERSION=1.4.309
23
+ UBUNTU_VERSION=jammy
24
+
23
25
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
25
27
apt-get update
26
28
apt-install-and-clear -y vulkan-sdk
You can’t perform that action at this time.
0 commit comments