2

I am currently working with PyTorch (more precisely with LSTMs using CUDA) on Ubuntu 18.04. As mentioned here, I have set CUBLAS_WORKSPACE_CONFIG=:4096:2.

However, if I train my LSTM using the same hyperparameters as before its performance decreases a lot. So I would like to reset the setting. Does anyone know the default value or how I could possibly obtain the settings I used before?

Hamed Hajiloo
  • 964
  • 1
  • 10
  • 31
Jacqueline
  • 21
  • 3

1 Answers1

0

I think to rest the environment variable you just need to unset the variable

simply in your terminal write

unset CUBLAS_WORKSPACE_CONFIG

for more info: How do I delete an exported environment variable?

Arij Aladel
  • 356
  • 1
  • 3
  • 10