python venv 虚拟环境配置 gpu

python的虚拟环境非常好用，尤其是python3之后，直接使用

python3 -m venv venvname

就可以创建虚拟环境。

source venvname/bin/activate

激活之后就可以直接使用。我遇到的问题是theano默认使用的版本是cpu, 虽然官网上提供了命令行运行的方式：

THANO_FLAGS='device=gpu0' python pythonfilename.py

也可以运行，但是如果想要在pycharm下调试的话就比较麻烦，所以想一步修改到位，直接修改虚拟环境当中的默认值。

首先修改的位置是

/venvname/lib/python2.7/site-packages/theano/configdefaults.py

python2.7处根据不同虚拟环境会改变。然后修改configdefaults.py文件里面的

AddConfigVar('device',
("Default device for computations. If gpu*, change the default to try "
"to move computation to it and to put shared variable of float32 "
"on it. Do not use upper case letters, only lower case even if "
"NVIDIA use capital letters."),
DeviceParam('cpu', allow_override=False),
in_c_key=False,
)

把‘cpu’修改为‘gpu0’即可，后面的0可以修改，看你电脑有多少gpu了。

修改之前，通过

import theano
print(theano.config)

看到的是这样的：



修改之后会变成：



这样一来，就会默认调用gpu了。