Commit
·
6d4a711
1
Parent(s):
48a6d6d
uh oh nightlies strike again
Browse files- Dockerfile +1 -1
Dockerfile
CHANGED
|
@@ -30,7 +30,7 @@ RUN set -eux; \
|
|
| 30 |
ENV LD_PRELOAD=/usr/local/cuda/lib64/libcusparse.so.12:/usr/local/cuda/lib64/libcublas.so.12:/usr/local/cuda/lib64/libcublasLt.so.12:/usr/local/cuda/lib64/libcufft.so.11:/usr/local/cuda/lib64/libcusolver.so.11
|
| 31 |
|
| 32 |
# Better allocator (less fragmentation than BFC during XLA autotune)
|
| 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
|
|
|
|
| 30 |
ENV LD_PRELOAD=/usr/local/cuda/lib64/libcusparse.so.12:/usr/local/cuda/lib64/libcublas.so.12:/usr/local/cuda/lib64/libcublasLt.so.12:/usr/local/cuda/lib64/libcufft.so.11:/usr/local/cuda/lib64/libcusolver.so.11
|
| 31 |
|
| 32 |
# Better allocator (less fragmentation than BFC during XLA autotune)
|
| 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
|