Pytorch使用指定的GPU
Posted 深度学习与PyTorch
tags:
篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了Pytorch使用指定的GPU相关的知识,希望对你有一定的参考价值。
Pytorch使用指定的GPU
如有问题或建议,请联系微信:
ymljdi
Pytorch默认从GPU0开始使用,如果GPU0正在运行其他程序,需要指定GPU
指定的方式有三种,
1.运行时指定
终端输入下面的指令
CUDA_VISIBLE_DEVICES=1 python train.py
2.代码中指定
import os
os.environ["CUDA_VISIBLE_DEVICES"] = "1"
3.使用函数set_device
import torch
torch.cuda.set_device(id)
以上是关于Pytorch使用指定的GPU的主要内容,如果未能解决你的问题,请参考以下文章
Pytorch - GPU ID 指定 pytorch gpu 指定