python虚拟环境--virtualenv
virtualenv 是一个创建隔绝的Python环境的工具。virtualenv创建一个包含所有必要的可执行文件的文件夹,用来使用Python工程所需的包。
- 为一个工程创建一个虚拟环境(实际创建一个文件夹):
fsj@octo:~$ virtualenv -p python3 n3lu/tensorflow
- 进入虚拟环境:需要被激活:
fsj@octo:~$ source ~/n3lu/tensorflow/bin/activate
- 安装tensorflow到虚拟环境 :
(tensorflow) fsj@octo:~$ pip install tensorflow-gpu
- 如果你在虚拟环境中暂时完成了工作,则可以停用它:
(tensorflow) fsj@octo:~$ deactivate
回到实际环境 fsj@octo:~$
- 在跑需要GPU的代码前,需指定GPU编号,否则会使用所有:
(tensorflow) fsj@octo:~$ export CUDA_VISIBLE_DEVICES=0,1
上面指定了仅可见 两块卡0,1
-
运行py脚本 python3 xxx.py
- 查看GPU信息及其进程:
(tensorflow) fsj@octo:~$ nvidia-smi
要删除一个虚拟环境,只需删除它的文件夹。(执行 rm -rf tensorflow
)。