Как говорится в документах TensorFlow, вы должны установить CUDA через pip, а затем пакет tensorflow, как он говорит. Затем перед активацией активируется
$ source ~/tensorflow/bin/activate # If using bash.
$ source ~/tensorflow/bin/activate.csh # If using csh.
(tensorflow)$ # Your prompt should change.
# Run Python programs that use TensorFlow.
...
# When you are done using TensorFlow, deactivate the environment.
(tensorflow)$ deactivate
Надеюсь, что это поможет.