Commit
·
22da097
1
Parent(s):
ca90a2d
updated docker
Browse files- Dockerfile +1 -1
Dockerfile
CHANGED
@@ -1,4 +1,4 @@
|
|
1 |
-
FROM nvidia/cuda:
|
2 |
|
3 |
# Install system dependencies for Python and build tools
|
4 |
RUN apt-get update && apt-get install -y \
|
|
|
1 |
+
FROM nvidia/cuda:12.1.1-devel-ubuntu22.04
|
2 |
|
3 |
# Install system dependencies for Python and build tools
|
4 |
RUN apt-get update && apt-get install -y \
|