sidoutcome commited on
Commit
a64b635
·
verified ·
1 Parent(s): ca7d80f

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +2 -2
Dockerfile CHANGED
@@ -1,5 +1,5 @@
1
- # Use NVIDIA CUDA base image with Python
2
- FROM nvidia/cuda:11.8.0-cudnn8-devel-ubuntu22.04 # ✅ Full cuDNN support
3
 
4
  # Install Python 3.10
5
  RUN apt-get update && apt-get install -y \
 
1
+ # Use NVIDIA CUDA base image with full cuDNN support
2
+ FROM nvidia/cuda:11.8.0-cudnn8-devel-ubuntu22.04
3
 
4
  # Install Python 3.10
5
  RUN apt-get update && apt-get install -y \