Spaces:
Running
on
A100
Running
on
A100
claude
Browse files- Dockerfile +1 -2
Dockerfile
CHANGED
|
@@ -35,8 +35,7 @@ RUN apt-get install -y --no-install-recommends \
|
|
| 35 |
libgl1-mesa-glx \
|
| 36 |
libpq-dev \
|
| 37 |
make \
|
| 38 |
-
cuda-toolkit-12
|
| 39 |
-
nvidia-cuda-toolkit
|
| 40 |
|
| 41 |
# Set CUDA environment variables
|
| 42 |
ENV CUDA_HOME=/usr/local/cuda
|
|
|
|
| 35 |
libgl1-mesa-glx \
|
| 36 |
libpq-dev \
|
| 37 |
make \
|
| 38 |
+
cuda-toolkit-12-2
|
|
|
|
| 39 |
|
| 40 |
# Set CUDA environment variables
|
| 41 |
ENV CUDA_HOME=/usr/local/cuda
|