Skip to content

Commit f2c0de0

Browse files
authored
Update linux_cuda_wheel.yaml
1 parent d9ff8bf commit f2c0de0

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

.github/workflows/linux_cuda_wheel.yaml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -84,6 +84,7 @@ jobs:
8484
echo cuda_version_without_periods=${cuda_version_without_periods} >> $GITHUB_ENV
8585
python_version_without_periods=$(echo "${{ matrix.python-version }}" | sed 's/\.//g')
8686
echo python_version_without_periods=${python_version_without_periods} >> $GITHUB_ENV
87+
echo LD_LIBRARY_PATH=$CONDA_PREFIX/lib:/usr/local/cuda/lib64/:${LD_LIBRARY_PATH} >> $GITHUB_ENV
8788
- uses: actions/download-artifact@v4
8889
with:
8990
name: meta-pytorch_torchcodec__${{ matrix.python-version }}_cu${{ env.cuda_version_without_periods }}_x86_64
@@ -140,7 +141,6 @@ jobs:
140141
ls
141142
- name: Run Python tests
142143
run: |
143-
export LD_LIBRARY_PATH=$CONDA_PREFIX/lib:/usr/local/cuda/lib64/:${LD_LIBRARY_PATH}
144144
${CONDA_RUN} FAIL_WITHOUT_CUDA=1 pytest --override-ini="addopts=-v" test --tb=short
145145
- name: Run Python benchmark
146146
run: |

0 commit comments

Comments
 (0)