fffiloni commited on
Commit
5c2db70
·
verified ·
1 Parent(s): bf3518b

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +1 -0
Dockerfile CHANGED
@@ -115,6 +115,7 @@ setup(\n\
115
  CUDAExtension(\n\
116
  name='_C',\n\
117
  sources=['/home/user/.local/lib/python3.10/site-packages/groundingdino/models/GroundingDINO/csrc/ms_deform_attn.cpp', '/home/user/.local/lib/python3.10/site-packages/groundingdino/models/GroundingDINO/csrc/ms_deform_attn_cuda.cu'],\n\
 
118
  ),\n\
119
  ],\n\
120
  cmdclass={\n\
 
115
  CUDAExtension(\n\
116
  name='_C',\n\
117
  sources=['/home/user/.local/lib/python3.10/site-packages/groundingdino/models/GroundingDINO/csrc/ms_deform_attn.cpp', '/home/user/.local/lib/python3.10/site-packages/groundingdino/models/GroundingDINO/csrc/ms_deform_attn_cuda.cu'],\n\
118
+ extra_compile_args={'cxx': ['-g'], 'nvcc': ['-O2', '-arch=sm_70']},\n\
119
  ),\n\
120
  ],\n\
121
  cmdclass={\n\