首页 技术 正文
技术 2022年11月14日
0 收藏 415 点赞 5,039 浏览 493 个字

转载请注明出处:

http://www.cnblogs.com/darkknightzh/p/8576825.html

如果使用多gpu运行程序,可以直接使用CUDA_VISIBLE_DEVICES=0,1,2,3 python xxx.py来设置该程序可见的gpu。当然也可以在程序开头设置os.environ[“CUDA_VISIBLE_DEVICES”] = ‘0,1,2,3’来使用。

如果在pycharm中调试时,使用多gpu的话,除了直接在程序中设置os.environ[“CUDA_VISIBLE_DEVICES”] = ‘0,1,2,3’,也可以通过下面的方法:

1. 打开Run/Debug Configurations,在Environment variables右边点击…处。

(原)pycharm中使用CUDA_VISIBLE_DEVICES

2. 点击下图中1处的’+’,会出现2处,增加CUDA_VISIBLE_DEVICES,再点击3处,填入0,1,2,3。(下图中第一行已经增加了,只不过为了说明,因而这样表示)

(原)pycharm中使用CUDA_VISIBLE_DEVICES

3. 在pycharm中正常运行程序,就可以了(为了测试是否正确,实际上使用的是2,3,4,5这4个GPU,显示的正确,说明这样配置有效果)。

(原)pycharm中使用CUDA_VISIBLE_DEVICES

下一篇: CV_Assert
相关推荐
python开发_常用的python模块及安装方法
adodb:我们领导推荐的数据库连接组件bsddb3:BerkeleyDB的连接组件Cheetah-1.0:我比较喜欢这个版本的cheeta…
日期:2022-11-24 点赞:878 阅读:9,492
Educational Codeforces Round 11 C. Hard Process 二分
C. Hard Process题目连接:http://www.codeforces.com/contest/660/problem/CDes…
日期:2022-11-24 点赞:807 阅读:5,907
下载Ubuntn 17.04 内核源代码
zengkefu@server1:/usr/src$ uname -aLinux server1 4.10.0-19-generic #21…
日期:2022-11-24 点赞:569 阅读:6,740
可用Active Desktop Calendar V7.86 注册码序列号
可用Active Desktop Calendar V7.86 注册码序列号Name: www.greendown.cn Code: &nb…
日期:2022-11-24 点赞:733 阅读:6,493
Android调用系统相机、自定义相机、处理大图片
Android调用系统相机和自定义相机实例本博文主要是介绍了android上使用相机进行拍照并显示的两种方式,并且由于涉及到要把拍到的照片显…
日期:2022-11-24 点赞:512 阅读:8,132
Struts的使用
一、Struts2的获取  Struts的官方网站为:http://struts.apache.org/  下载完Struts2的jar包,…
日期:2022-11-24 点赞:671 阅读:5,295