File tree Expand file tree Collapse file tree 2 files changed +0
-9
lines changed
TL0_python-self-test-base-cuda Expand file tree Collapse file tree 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" && \
1212 ln -s cuda-12.0 /usr/local/cuda
1313version_ge " $DALI_CUDA_MAJOR_VERSION " " 11" && \
1414 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` \
1815 ` pip list | grep nvidia-npp | cut -d " " -f1` \
1916 || true
2017
@@ -42,7 +39,4 @@ version_eq "$DALI_CUDA_MAJOR_VERSION" "11" && \
4239version_ge " $DALI_CUDA_MAJOR_VERSION " " 11" && \
4340 pip install nvidia-cufft-cu${DALI_CUDA_MAJOR_VERSION} \
4441 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} \
4842 || true
Original file line number Diff line number Diff line change 159159 if [ " $( uname -m) " == " x86_64" ] && [ -z " ${DO_NOT_INSTALL_CUDA_WHEEL} " ] && [ -z " ${CONDA_PREFIX} " ]; then
160160 NPP_VERSION=$( if [[ $DALI_CUDA_MAJOR_VERSION == " 12" ]]; then echo " ==12.2.5.30" ; else echo " " ; fi)
161161 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} \
165162 nvidia-cufft-cu${DALI_CUDA_MAJOR_VERSION} \
166163 -f /pip-packages"
167164 fi
You can’t perform that action at this time.
0 commit comments