use nvidia runtime for tests · ZahidSQLDBA/docker-python@c170f0a · GitHub

ZahidSQLDBA / docker-python Public

forked from Kaggle/docker-python

• Notifications You must be signed in to change notification settings
• Fork 0
• Star 0

• Code
• Pull requests
• Actions
• Projects
• Security and quality
• Insights

Additional navigation options

Commit c170f0a

Browse filesBrowse the repository at this point in the historyBrowse files

djherbis

committed

use nvidia runtime for tests

1 parent ffa7d0d commit c170f0a
Copy full SHA for c170f0a

1 file changed

+1-1Lines changed: 1 addition & 1 deletion

File tree

Expand file treeCollapse file tree

Open diff view settings

Filter options

• test

Expand file treeCollapse file tree

Open diff view settings

Collapse file

‎test‎

Copy file name to clipboardExpand all lines: test

+1-1Lines changed: 1 addition & 1 deletion

Original file line number	Diff line number	Diff line change
@@ -28,7 +28,7 @@ while :; do
28	28	;;
29	29	-g|--gpu)
30	30	IMAGE_TAG='kaggle/python-gpu-build'
31		- ADDITONAL_OPTS='-v /tmp/empty_dir:/usr/local/cuda/lib64/stubs:ro'
	31	+ ADDITONAL_OPTS='--runtime nvidia -v /tmp/empty_dir:/usr/local/cuda/lib64/stubs:ro'
32	32	;;
33	33	-i|--image)
34	34	if [[ -z $2 ]]; then

0 commit comments

Comments

0 (0)