diff --git a/scripts/larvatagger-toolkit.jl b/scripts/larvatagger-toolkit.jl
index 08988999ad27cc2aba2da750f152c052887d12cc..6ba92b70044f5a2f3619e4f479f613b4f4c5e0b2 100755
--- a/scripts/larvatagger-toolkit.jl
+++ b/scripts/larvatagger-toolkit.jl
@@ -9,6 +9,8 @@ fi
 FLAGS=
 if [ "$1" = "--sysimage" -o "$1" = "-J" ]; then FLAGS="--sysimage $2 "; shift 2; fi
 if [ "${1:0:2}" = "-J" ]; then FLAGS="$1"; shift; fi
+if [ "$1" = train ] && [ -n "$(which nvidia-smi)" ] && [ -z "$CUBLAS_WORKSPACE_CONFIG" ];
+then export CUBLAS_WORKSPACE_CONFIG=:4096:8; fi
 if [ -z "$JULIA" ]; then JULIA=julia; fi
 exec $JULIA --project="$PROJECT_DIR" --color=yes --startup-file=no $FLAGS\
     "${BASH_SOURCE[0]}" "$@"
diff --git a/test/scenarii.sh b/test/scenarii.sh
index 2f577b43fb828e6c5fdf2b6dc9062ca3afe523c9..5f8932ac7d3709e40f950be4ec1bdc84d585dca6 100755
--- a/test/scenarii.sh
+++ b/test/scenarii.sh
@@ -18,11 +18,11 @@ if [ -z "$JULIA_PROJECT" ]; then
 fi
 
 if [ -n "$(which nvidia-smi)" ]; then
-  if [ -z "$CUBLAS_WORKSPACE_CONFIG" ]; then
-    echo "Setting CUBLAS_WORKSPACE_CONFIG=:4096:8 to ensure reproducibility;"
-    echo "see https://docs.nvidia.com/cuda/cublas/index.html#results-reproducibility"
-    export CUBLAS_WORKSPACE_CONFIG=:4096:8
-  fi
+  # if [ -z "$CUBLAS_WORKSPACE_CONFIG" ]; then
+  #   echo "Setting CUBLAS_WORKSPACE_CONFIG=:4096:8 to ensure reproducibility;"
+  #   echo "see https://docs.nvidia.com/cuda/cublas/index.html#results-reproducibility"
+  #   export CUBLAS_WORKSPACE_CONFIG=:4096:8
+  # fi
   tagger_suffix=_gpu
 fi