python venv 虚拟环境配置 gpu
程序员文章站
2022-06-06 17:07:03
...
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了。
下一篇: 赵幽缪王为什么会那么信任李牧?