Skip to content

Commit 25e1859

Browse files
committed
Enable nvidia container runtime on oracle 8 dockerfiles
1 parent 4fdf8a8 commit 25e1859

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

tools/imagesets/oracle8conda/runtime/Dockerfile

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -38,3 +38,7 @@ RUN yum install -y \
3838
libcufft-$CUDA_PKG_VERSION \
3939
&& yum clean all \
4040
&& rm -rf /var/cache/yum
41+
42+
# nvidia-container-runtime
43+
ENV NVIDIA_VISIBLE_DEVICES all
44+
ENV NVIDIA_DRIVER_CAPABILITIES compute,utility

0 commit comments

Comments
 (0)