fffiloni commited on
Commit
bf2b89f
·
1 Parent(s): d79a11e

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +1 -1
Dockerfile CHANGED
@@ -1,5 +1,5 @@
1
  # Use an official PyTorch image with CUDA support as the base image
2
- FROM pytorch/pytorch:1.9.01-cuda11.1-cudnn8-runtime
3
 
4
  # Install Git, OpenGL libraries, and libglib2.0
5
  RUN apt-get update && apt-get install -y git libgl1-mesa-glx libglib2.0-0
 
1
  # Use an official PyTorch image with CUDA support as the base image
2
+ FROM pytorch/pytorch:1.9.0-cuda11.1-cudnn8-runtime
3
 
4
  # Install Git, OpenGL libraries, and libglib2.0
5
  RUN apt-get update && apt-get install -y git libgl1-mesa-glx libglib2.0-0