File tree 2 files changed +0
-9
lines changed
TL0_python-self-test-base-cuda
2 files changed +0
-9
lines changed Original file line number Diff line number Diff line change @@ -12,9 +12,6 @@ version_eq "$DALI_CUDA_MAJOR_VERSION" "12" && \
12
12
ln -s cuda-12.0 /usr/local/cuda
13
13
version_ge " $DALI_CUDA_MAJOR_VERSION " " 11" && \
14
14
pip uninstall -y ` pip list | grep nvidia-cufft | cut -d " " -f1` \
15
- ` pip list | grep nvidia-nvjpeg | cut -d " " -f1` \
16
- ` pip list | grep nvidia-nvjpeg2k | cut -d " " -f1` \
17
- ` pip list | grep nvidia-nvtiff | cut -d " " -f1` \
18
15
` pip list | grep nvidia-npp | cut -d " " -f1` \
19
16
|| true
20
17
@@ -42,7 +39,4 @@ version_eq "$DALI_CUDA_MAJOR_VERSION" "11" && \
42
39
version_ge " $DALI_CUDA_MAJOR_VERSION " " 11" && \
43
40
pip install nvidia-cufft-cu${DALI_CUDA_MAJOR_VERSION} \
44
41
nvidia-npp-cu${DALI_CUDA_MAJOR_VERSION} \
45
- nvidia-nvjpeg-cu${DALI_CUDA_MAJOR_VERSION} \
46
- nvidia-nvjpeg2k-cu${DALI_CUDA_MAJOR_VERSION} \
47
- nvidia-nvtiff-cu${DALI_CUDA_MAJOR_VERSION} \
48
42
|| true
Original file line number Diff line number Diff line change 159
159
if [ " $( uname -m) " == " x86_64" ] && [ -z " ${DO_NOT_INSTALL_CUDA_WHEEL} " ] && [ -z " ${CONDA_PREFIX} " ]; then
160
160
NPP_VERSION=$( if [[ $DALI_CUDA_MAJOR_VERSION == " 12" ]]; then echo " ==12.2.5.30" ; else echo " " ; fi)
161
161
install_pip_pkg " pip install --upgrade nvidia-npp-cu${DALI_CUDA_MAJOR_VERSION}${NPP_VERSION} \
162
- nvidia-nvjpeg-cu${DALI_CUDA_MAJOR_VERSION} \
163
- nvidia-nvjpeg2k-cu${DALI_CUDA_MAJOR_VERSION} \
164
- nvidia-nvtiff-cu${DALI_CUDA_MAJOR_VERSION} \
165
162
nvidia-cufft-cu${DALI_CUDA_MAJOR_VERSION} \
166
163
-f /pip-packages"
167
164
fi
You can’t perform that action at this time.
0 commit comments