File tree Expand file tree Collapse file tree 1 file changed +3
-1
lines changed Expand file tree Collapse file tree 1 file changed +3
-1
lines changed Original file line number Diff line number Diff line change @@ -37,11 +37,13 @@ RUN if [ "${BUILD_TYPE}" = "cublas" ]; then \
37
37
apt-get update && \
38
38
apt-get install -y cuda-nvcc-${CUDA_MAJOR_VERSION}-${CUDA_MINOR_VERSION} libcublas-dev-${CUDA_MAJOR_VERSION}-${CUDA_MINOR_VERSION} libcusparse-dev-${CUDA_MAJOR_VERSION}-${CUDA_MINOR_VERSION} libcusolver-dev-${CUDA_MAJOR_VERSION}-${CUDA_MINOR_VERSION} && apt-get clean \
39
39
; fi
40
+
40
41
# oneapi requirements
41
42
RUN if [ "${BUILD_TYPE}" = "sycl_f16" ] || [ "${BUILD_TYPE}" = "sycl_f32" ]; then \
42
43
wget https://registrationcenter-download.intel.com/akdlm/IRC_NAS/163da6e4-56eb-4948-aba3-debcec61c064/l_BaseKit_p_2024.0.1.46_offline.sh && \
43
- sh ./l_BaseKit_p_2024.0.1.46_offline.sh \
44
+ sh ./l_BaseKit_p_2024.0.1.46_offline.sh -a -s --eula accept \
44
45
; fi
46
+
45
47
ENV PATH /usr/local/cuda/bin:${PATH}
46
48
47
49
# OpenBLAS requirements and stable diffusion
You can’t perform that action at this time.
0 commit comments