python虚拟环境运行TensorFlow

python虚拟环境--virtualenv

virtualenv 是一个创建隔绝的Python环境的工具。virtualenv创建一个包含所有必要的可执行文件的文件夹,用来使用Python工程所需的包。

  1. 为一个工程创建一个虚拟环境(实际创建一个文件夹):
    fsj@octo:~$ virtualenv -p python3 n3lu/tensorflow
  2. 进入虚拟环境:需要被激活:
    fsj@octo:~$ source ~/n3lu/tensorflow/bin/activate
  3. 安装tensorflow到虚拟环境 :
    (tensorflow) fsj@octo:~$ pip install tensorflow-gpu
  4. 如果你在虚拟环境中暂时完成了工作,则可以停用它:
    (tensorflow) fsj@octo:~$ deactivate

    回到实际环境  fsj@octo:~$

  5. 在跑需要GPU的代码前,需指定GPU编号,否则会使用所有:
    (tensorflow) fsj@octo:~$ export CUDA_VISIBLE_DEVICES=0,1

    上面指定了仅可见 两块卡0,1

  6. 运行py脚本  python3 xxx.py

  7. 查看GPU信息及其进程:
    (tensorflow) fsj@octo:~$ nvidia-smi

要删除一个虚拟环境,只需删除它的文件夹。(执行 rm -rf tensorflow )。

猜你喜欢

转载自blog.csdn.net/qq_22930277/article/details/81077358