File tree Expand file tree Collapse file tree 2 files changed +6
-5
lines changed Expand file tree Collapse file tree 2 files changed +6
-5
lines changed Original file line number Diff line number Diff line change 7
7
- main
8
8
paths :
9
9
- ' docker/**'
10
+ - ' .github/workflows/release-ci-docker.yml'
10
11
pull_request :
11
12
branches :
12
13
- main
Original file line number Diff line number Diff line change @@ -24,15 +24,15 @@ CUDA_VERSION=${1:-cu128}
24
24
25
25
pip3 install torch --index-url https://download.pytorch.org/whl/${CUDA_VERSION}
26
26
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 '
28
28
pip3 install nvidia-cutlass-dsl
29
29
pip3 install ' nvidia-cudnn-frontend>=1.13.0'
30
30
31
31
# Install cudnn package based on CUDA version
32
32
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"
35
35
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"
38
38
fi
You can’t perform that action at this time.
0 commit comments