fffiloni commited on
Commit
4dde7d9
·
1 Parent(s): fbb2bee

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +5 -2
Dockerfile CHANGED
@@ -1,8 +1,11 @@
1
  # Use an official PyTorch image with CUDA support as the base image
2
  FROM pytorch/pytorch:1.12.1-cuda11.3-cudnn8-runtime
3
 
4
- # Install Git and OpenGL libraries
5
- RUN apt-get update && apt-get install -y git libgl1-mesa-glx
 
 
 
6
 
7
  RUN apt-get update && apt-get install -y ninja-build
8
 
 
1
  # Use an official PyTorch image with CUDA support as the base image
2
  FROM pytorch/pytorch:1.12.1-cuda11.3-cudnn8-runtime
3
 
4
+ # Install Triton Inference Server
5
+ RUN pip install tritonclient[http]
6
+
7
+ # Install Git and OpenGL libraries, and libglib2.0
8
+ RUN apt-get update && apt-get install -y git libgl1-mesa-glx libglib2.0-0
9
 
10
  RUN apt-get update && apt-get install -y ninja-build
11