当前位置:首页|资讯

「深度学习」Python使用指定gpu运行代码

作者:小松时尚秀吧发布时间:2024-12-24

命令行指定显卡GPU运行python脚本

第二种方式:在python脚本文件中添加如下内容进行指定:

如果在脚本文件中已经写明了第二种方式,则不支持第一种方式的指定。 2、指定使用多张显卡运行脚本在GPU的id为0和1的两张显卡上运行***.py程序:

3、在单张显卡的情况下开启多个进程运行脚本命令行程序如下:(其中NUM_GPUS_YOU_HAVE代表运行的进程数)

说明:程序会根据卡的容量分配成两个程序,不一定在两张卡,有时候一张卡的容量够则会在该张卡上开启新的进程。

  1. 查看GPU状态

  2. 1、单次查看

  1. 隔一秒查看GPU状态:

使用指定gpu运行代码

一、前提

1、在命令行使用nvidia-smi查看gpu设备情况,当存在空闲设备时才能用,否则会出现运行内存不够出错的情况(具体参考文章GPU之nvidia-smi命令详解); 2、安装好cuda和cudcnn(具体步骤请参考:①windows:CUDA安装教程(超详细)),②linux:linux安装CUDA+cuDNN) 在命令行输入nvcc -V或者nvcc --version检查是否安装成功:

二、命令行直接指定

运行python文件时,在前面加上 CUDA_VISIBLE_DEVICES=xxx,xxx是你要指定的gpu序号。如CUDA_VISIBLE_DEVICES=0,2 python extract_masks.py表示指定了0、2号两个gpu。注意,序号是从0开始排的,所以如果你只有一个gpu,那么就是CUDA_VISIBLE_DEVICES=0 python extract_masks.py。

三、在python文件中指定

在需要运行的python文件最开始加上两行如下,等号后面的数字同上所述。

四、如何使用

python文件中,定义需要加速的模型之后,加上:


Copyright © 2024 aigcdaily.cn  北京智识时代科技有限公司  版权所有  京ICP备2023006237号-1