以下是一个示例代码，可以根据用户选择安装tensorflow或tensorflow-gpu：

import os
import sys

def install_tensorflow():
    os.system('pip install tensorflow')

def install_tensorflow_gpu():
    os.system('pip install tensorflow-gpu')

def main():
    choice = input("请选择要安装的版本（输入1安装tensorflow，输入2安装tensorflow-gpu）: ")
    if choice == '1':
        install_tensorflow()
    elif choice == '2':
        install_tensorflow_gpu()
    else:
        print("无效的选择，请重新运行程序并输入有效的选择。")
        sys.exit(1)

if __name__ == '__main__':
    main()

运行以上代码，用户将被要求输入1或2来选择要安装的版本。根据用户的选择，代码将使用pip命令安装相应的tensorflow版本。