cuda

Dockerfile

@@ -8,7 +8,7 @@ COPY ./requirements.txt /code/requirements.txt
 #https://download.pytorch.org/whl/torch_stable.ht
 
 #https://stackoverflow.com/questions/55313610/importerror-libgl-so-1-cannot-open-shared-object-file-no-such-file-or-directo
-RUN apt-get update && apt-get install python3.9 python3-pip ffmpeg libsm6 libxext6  -y
+RUN apt-get update && apt-get install python3.9 python3-pip git ffmpeg libsm6 libxext6  -y
 
 RUN pip install torch --extra-index-url https://download.pytorch.org/whl/cu113
 # --index-url https://download.pytorch.org/whl/cu117

main.py

@@ -1,5 +1,3 @@
-
-
 # Pose inferencing
 import mmpose
 from mmpose.apis import MMPoseInferencer
@@ -29,6 +27,8 @@ if torch.cuda.is_available():
 else:
     device = torch.device("cpu")
 
+os.system("nvidia-smi")
+
 print("[INFO]: Imported modules!")
 human = MMPoseInferencer("human")
 hand = MMPoseInferencer("hand")