pk10平台租用(企 娥:217 1793 408)是一个创建隔绝的Python环境的工具。virtualenv创建一个包含所有必要的可执行文件的文件夹,用来使用Python工程所需的包。
为一个工程创建一个虚拟环境(实际创建一个文件夹):
[email protected]:~$ virtualenv -p python3 n3lu/tensorflow
进入虚拟环境:需要被激活:
[email protected]:~$ source ~/n3lu/tensorflow/bin/activate
安装tensorflow到虚拟环境 :
(tensorflow) [email protected]:~$ pip install tensorflow-gpu
如果你在虚拟环境中暂时完成了工作,则可以停用它:
(tensorflow) [email protected]:~$ deactivate
回到实际环境 [email protected]:~$
在跑需要GPU的代码前,需指定GPU编号,否则会使用所有:
(tensorflow) [email protected]:~$ export CUDA_VISIBLE_DEVICES=0,1
上面指定了仅可见 两块卡0,1
运行py脚本 python3 xxx.py
查看GPU信息及其进程:
(tensorflow) [email protected]:~$ nvidia-smi
原文地址:http://blog.51cto.com/13874313/2145821
时间: 2024-11-13 07:05:32