欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

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)

看到的是这样的:
python venv 虚拟环境配置 gpu

修改之后会变成:
python venv 虚拟环境配置 gpu

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

相关标签: python 虚拟环境