Skip to content
This repository has been archived by the owner on Nov 14, 2023. It is now read-only.

Commit

Permalink
pth-facebookresearch-sam-vit-h: add NVIDIA environment variables to t…
Browse files Browse the repository at this point in the history
…he GPU image (cvat-ai#6171)

Without these variables, the container cannot actually use CUDA. At
least that was my experience on AWS; it's possible that in other
configurations the defaults are different.

See
<https://docs.nvidia.com/datacenter/cloud-native/container-toolkit/user-guide.html>.
  • Loading branch information
SpecLad authored and mikhail-treskin committed Jul 1, 2023
1 parent d5b268d commit 116d7e7
Showing 1 changed file with 5 additions and 0 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -31,6 +31,11 @@ spec:

directives:
preCopy:
# set NVIDIA container runtime settings
- kind: ENV
value: NVIDIA_VISIBLE_DEVICES=all
- kind: ENV
value: NVIDIA_DRIVER_CAPABILITIES=compute,utility
# disable interactive frontend
- kind: ENV
value: DEBIAN_FRONTEND=noninteractive
Expand Down

0 comments on commit 116d7e7

Please sign in to comment.