ikuinen99 commited on
Commit
9a31057
·
1 Parent(s): b0359a0
Files changed (2) hide show
  1. Dockerfile +2 -0
  2. requirements.txt +1 -0
Dockerfile ADDED
@@ -0,0 +1,2 @@
 
 
 
1
+ ENV CUDA_HOME /usr/local/cuda-11.3/
2
+ RUN pip install git+https://github.com/IDEA-Research/GroundingDINO.git
requirements.txt CHANGED
@@ -1,3 +1,4 @@
 
1
  accelerate
2
  aiohttp==3.8.4
3
  aiosignal==1.3.1
 
1
+ --extra-index-url https://download.pytorch.org/whl/cu113
2
  accelerate
3
  aiohttp==3.8.4
4
  aiosignal==1.3.1