Skip to content

Commit ef9dd60

Browse files
authored
docker: upgrade apache-tvm-ffi==0.1.0b15 in docker container (#1857)
1 parent c87f962 commit ef9dd60

File tree

2 files changed

+6
-5
lines changed

2 files changed

+6
-5
lines changed

.github/workflows/release-ci-docker.yml

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -7,6 +7,7 @@ on:
77
- main
88
paths:
99
- 'docker/**'
10+
- '.github/workflows/release-ci-docker.yml'
1011
pull_request:
1112
branches:
1213
- main

docker/install/install_python_packages.sh

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -24,15 +24,15 @@ CUDA_VERSION=${1:-cu128}
2424

2525
pip3 install torch --index-url https://download.pytorch.org/whl/${CUDA_VERSION}
2626
pip3 install requests responses ninja pytest numpy scipy build nvidia-ml-py cuda-python einops nvidia-nvshmem-cu12
27-
pip3 install 'apache-tvm-ffi==0.1.0b11'
27+
pip3 install 'apache-tvm-ffi==0.1.0b15'
2828
pip3 install nvidia-cutlass-dsl
2929
pip3 install 'nvidia-cudnn-frontend>=1.13.0'
3030

3131
# Install cudnn package based on CUDA version
3232
if [[ "$CUDA_VERSION" == *"cu13"* ]]; then
33-
pip3 install --upgrade cuda-python==13.0
34-
pip3 install "nvidia-cudnn-cu13>=9.12.0.46"
33+
pip3 install --upgrade cuda-python==13.0
34+
pip3 install "nvidia-cudnn-cu13>=9.12.0.46"
3535
else
36-
pip3 install --upgrade cuda-python==12.*
37-
pip3 install "nvidia-cudnn-cu12>=9.11.0.98"
36+
pip3 install --upgrade cuda-python==12.*
37+
pip3 install "nvidia-cudnn-cu12>=9.11.0.98"
3838
fi

0 commit comments

Comments
 (0)