(原)PyTorch中使用指定的GPU
PyTorch 指定 GPU 使用
2023-09-11 14:22:24 时间
转载请注明出处:
http://www.cnblogs.com/darkknightzh/p/6836568.html
PyTorch默认使用从0开始的GPU,如果GPU0正在运行程序,需要指定其他GPU。
有如下两种方法来指定需要使用的GPU。
1. 类似tensorflow指定GPU的方式,使用CUDA_VISIBLE_DEVICES。
1.1 直接终端中设定:
CUDA_VISIBLE_DEVICES=1 python my_script.py
1.2 python代码中设定:
import os
os.environ["CUDA_VISIBLE_DEVICES"] = "2"
见网址:http://www.cnblogs.com/darkknightzh/p/6591923.html
2. 使用函数 set_device
import torch torch.cuda.set_device(id)
该函数见 pytorch-master\torch\cuda\__init__.py。
不过官方建议使用CUDA_VISIBLE_DEVICES,不建议使用 set_device 函数。
相关文章
- 【PyTorch】教程:torch.nn.Softshrink
- 【PyTorch】教程:学习基础知识-(3) Datasets-DataLoader
- jittor和pytorch 生成网络对比之clustergan
- pytorch使用horovod多gpu训练
- 【图像分类】——来来来,干了这碗EfficientNet实战(Pytorch)
- 【图像分类】实战——使用ResNet实现猫狗分类(pytorch)
- pytorch Dropout 使用
- PyTorch查看模型和数据是否在GPU上
- PyTorch学习笔记(二):PyTorch简介与基础知识
- 【PyTorch】 多进程队列中传入pytorch处理后的tensor张量
- 【深度学习】Pytorch面试题:什么是 PyTorch?PyTorch 的基本要素是什么?Conv1d、Conv2d 和 Conv3d 有什么区别?
- (5)OpenPose - Pytorch版用于人体姿态检测、关键点提取、摔倒检测
- 【Pytorch】第 5 章 :解决多臂老虎机问题
- 【Pytorch】第 1 章 :强化学习和 PyTorch 入门
- 【Pytorch Lighting】第 6 章:深度生成模型
- pytorch 32 onnx模型每次输出的结果都会变的解决方案