如何在深度学习中用指定的GPU来跑代码

1.在机子有多张显卡的情况下，可以使用 nvidia-smi查看所有显卡的编号及使用情况 ，如下：

2.可以看到，笔者的机子总共有两张显卡，分别是GPU0和GPU1。其中，显卡0相对来说更为空闲。

3.因此，笔者可以采用如下语句指定GPU0来跑代码：

import os
os.environ["CUDA_VISIBLE_DEVICES"] = "0"

该语句一般放在main.py的最开始部分，大家根据自己的情况进行修改即可。

如何在深度学习中用指定的GPU来跑代码相关推荐

用MXnet实战深度学习之一：安装GPU版mxnet并跑一个MNIST手写数字识别（zz）
用MXnet实战深度学习之一:安装GPU版mxnet并跑一个MNIST手写数字识别我想写一系列深度学习的简单实战教程,用mxnet做实现平台的实例代码简单讲解深度学习常用的一些技术方向和实战样例.这 ...
win11动手学深度学习安装过程（GPU版）（ CUDA Anaconda Mxnet )
win11,动手学深度学习安装过程(GPU版)( CUDA / Anaconda / Mxnet )(备忘) 一.检查及安装CUDA 首先检查显卡支持的cuda: 直接搜索NVIDIA,选择NVIDI ...
深度学习训练吃显卡_在深度学习中喂饱GPU
原标题:在深度学习中喂饱GPU 新智元推荐来源:知乎专栏作者:风车车 [新智元导读]深度学习模型训练是不是大力出奇迹,显卡越多越好?非也,没有512张显卡,也可以通过一些小技巧优化模型训练.本文作 ...
【翻译】为深度学习购买哪种GPU: 我在深度学习中使用GPU的经验和建议
文章目录概述 GPU是如何工作的? 对深度学习处理速度最重要的GPU规格张量核心没有张量核心的矩阵乘法使用张量核心的矩阵乘法带有张量核心和异步拷贝的矩阵乘法(RTX 30/RTX 40)和T ...
深度学习相关的主要 GPU 初识
和深度学习相关的主要 GPU 性能指标如下: 内存带宽:GPU 处理大量数据的能力,是最重要的性能指标. 处理能力:表示 GPU 处理数据的速度,将其量化为 CUDA 核心数量和每一个核心的频率的乘积 ...
【深度学习】谷歌云GPU服务器创建与使用指南（二）
接上一篇:[深度学习]谷歌云GPU服务器创建与使用指南(一) 这篇介绍ssh连接及一款客户端xshell 本篇将分为以下几点讲述: 1.ssh客户端 2.ssh公钥配置 3.使用xshell连接ssh ...
【动手学深度学习PyTorch版】23 深度学习硬件CPU 和 GPU
上一篇请移步[动手学深度学习PyTorch版]22续 ResNet为什么能训练出1000层的模型_水w的博客-CSDN博客目录一.深度学习硬件CPU 和 GPU 1.1 深度学习硬件 ◼ 计算机构 ...
python 深度学习模型训练多GPU下如何调用
原文链接<pytorch指定多块GPU运行代码>, 一起来围观吧 https://blog.csdn.net/OneQuestionADay/article/details/1116914 ...
【AI超级美发师】深度学习算法打造染发特效（附代码）
[新智元导读]如今,在类似天天P图.美图秀秀等手机APP中,给指定照片或视频中的人物更换头发颜色已经是再正常不过的事情了.那么本文便介绍了该功能背后如AI头发分割模块.头发换色.颜色增强与修正模块等技 ...

如何在深度学习中用指定的GPU来跑代码

如何在深度学习中用指定的GPU来跑代码相关推荐

最新文章

热门文章