Commit
·
48a6d6d
1
Parent(s):
de5d1e6
reverting
Browse files- Dockerfile +1 -1
Dockerfile
CHANGED
|
@@ -33,7 +33,7 @@ ENV LD_PRELOAD=/usr/local/cuda/lib64/libcusparse.so.12:/usr/local/cuda/lib64/lib
|
|
| 33 |
ENV TF_GPU_ALLOCATOR=cuda_malloc_async
|
| 34 |
|
| 35 |
# Let cuBLAS use TF32 fast path on Ada (L40S) for big GEMMs
|
| 36 |
-
ENV TF_ENABLE_CUBLAS_TF32=1 NVIDIA_TF32_OVERRIDE=1
|
| 37 |
|
| 38 |
ENV DEBIAN_FRONTEND=noninteractive \
|
| 39 |
PYTHONUNBUFFERED=1 \
|
|
|
|
| 33 |
ENV TF_GPU_ALLOCATOR=cuda_malloc_async
|
| 34 |
|
| 35 |
# Let cuBLAS use TF32 fast path on Ada (L40S) for big GEMMs
|
| 36 |
+
# ENV TF_ENABLE_CUBLAS_TF32=1 NVIDIA_TF32_OVERRIDE=1
|
| 37 |
|
| 38 |
ENV DEBIAN_FRONTEND=noninteractive \
|
| 39 |
PYTHONUNBUFFERED=1 \
|