torch.cuda.set_device(device)

已于 2023-11-02 20:19:49 修改
阅读量1.8w 收藏 2
点赞数 4
分类专栏： Pytorch 文章标签： 计算机视觉 深度学习 机器学习
于 2022-05-08 09:17:16 首次发布
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.csdn.net/weixin_36670529/article/details/105969442
版权
此篇博客建议避免使用torch.cuda.set_device()函数，转而采用环境变量CUDA_VISIBLE_DEVICES设置当前设备。它详细解释了为什么推荐这种改变，并提到了在大多数情况下，如何通过环境变量来更有效地管理CUDA设备。
摘要由CSDN通过智能技术生成

参考  torch.cuda - 云+社区 - 腾讯云

torch.cuda.set_device(device)[source]

Sets the current device.

Usage of this function is discouraged in favor of device. In most cases it’s better to use CUDA_VISIBLE_DEVICES environmental variable.

Parameters：

  • device (torch.device or int) – selected device. This function is a no-op if this argument is negative.

Wanderer001
关注
专栏目录
Pytorch基础：torch.cuda.set_device函数
前进，才知道自己的渺小
05-13 2223
torch.cuda.set_device函数用于设置当前使用的cuda设备，在当拥有多个可用的GPU且能被pytorch识别的cuda设备情况下（环境变量CUDA_VISIBLE_DEVICES可以影响GPU设备到cuda设备的映射）。
关于AttributeError: module ‘torch._C‘ has no attribute ‘_cuda_setDevice‘问题的解决
热门推荐
qq_39578356的博客
09-14 2万+
1.该问题错误情况如下 File "E:\PythonCode\STANet-master\options\base_options.py", line 140, in parse torch.cuda.set_device(opt.gpu_ids[0]) File "E:\Python\lib\site-packages\torch\cuda\__init__.py", line 281, in set_device torch._C._cuda_setDevice(device)
torch.cuda.set_divice()
最新发布
qq_45759229的博客
08-21 236
我申请了两块GPU,然后看两张显卡的编号。
torch.device和torch.cuda.set_device()
weixin_42559117的博客
10-29 4578
2. 如果使用torch.cuda.set_device('cuda:0')的方式，则改用。3. 即前者需要采用to()方法，而后者可以使用cuda()方法，而无需传递参数。1. 使用torch.device的方式设置显卡之后，需用使用。的方式将模型和数据放到GPU上。的方式将模型和数据加载到显卡中。
torch.cuda.set_device(6)
dantadanta的博客
12-19 4154
选择显卡LINUX
【设置gpu设备】os.environ[‘CUDA_VISIBLE_DEVICES‘] 和 torch.cuda.set_device()
Danger的博客
03-23 1万+
【设置gpu设备】os.environ[‘CUDA_VISIBLE_DEVICES‘] 和 torch.cuda.set_device()
pytorch的torch.cuda.set_per_process_memory_fraction()函数介绍
weixin_42993916的博客
03-11 6270
函数功能： 在pytorch中设置显存使用比例，即能够完成显存的使用上限设置。 torch.cuda.set_per_process_memory_fraction(0.5, 0) 参数1：fraction 限制的上限比例，如0.5 就是总GPU显存的一半，可以是0~1的任意float大小； 参数2：device 设备号； 如0 表示GPU卡 0号； 功能解释如下： Set memory fraction for a process. The fraction is used to l
torch.cuda.set_device()方法
07-28
引用\[1\]和\[2\]介绍了使用torch.cuda.set_device()方法来指定使用的显卡。这个方法可以用来设置主显卡或者指定特定的显卡。在使用这个方法之前，需要先导入torch库。然后可以使用torch.cuda.set_device()方法来...
ubuntu安装python3-vene_Python torch.set_default_tensor_type方法代码示例
weixin_39966465的博客
11-22 713
本文整理汇总了Python中torch.set_default_tensor_type方法的典型用法代码示例。如果您正苦于以下问题：Python torch.set_default_tensor_type方法的具体用法？Python torch.set_default_tensor_type怎么用？Python torch.set_default_tensor_type使用的例子？那么恭喜您, 这...
RuntimeError: Attempting to deserialize object on a CUDA device but torch.cuda.is_available() is Fal
Hello Word!
07-19 6493
在跑Pytorch模型测试代码时报错: Traceback (most recent call last): File "../models/arc_face.py", line 44, in Arcface learner.load_state(conf, 'ir_se50.pth', model_only=True, from_save_folder=False) File "../arcface/Learner.py", line 86, in load_state pretr
torch.cuda
weixin_36670529的博客
08-30 8509
目录 torch.cuda.current_blas_handle() torch.cuda.current_device() torch.cuda.current_stream(device=None) torch.cuda.default_stream(device=None) class torch.cuda.device(device) torch.cuda.device_co...
[转载]pytorch查看gpu信息
Xavier的博客
04-08 388
为什么将数据转移至GPU的方法叫做.cuda而不是.gpu，就像将数据转移至CPU调用的方法是.cpu？这是因为GPU的编程接口采用CUDA，而目前并不是所有的GPU都支持CUDA，只有部分Nvidia的GPU才支持。PyTorch未来可能会支持AMD的GPU，而AMD GPU的编程接口采用OpenCL，因此PyTorch还预留着.cl方法，用于以后支持AMD等的GPU。 torch.cuda.is_available() cuda是否可用； torch.cuda.device_count() 返回gpu
os.environ[‘CUDA_VISIBLE_DEVICES‘] 和torch.cuda.set_device()
weixin_46141646的博客
02-15 3643
在使用 PyTorch 进行训练之前，需要确保已经正确设置了可见的 GPU 设备，并且已经初始化了 CUDA 环境。是一个环境变量，可以通过设置它来限制程序所能看到的可用 GPU 设备列表，从而确保程序只使用指定的 GPU 设备。设置该环境变量可以使用。则是一个 PyTorch 提供的函数，用于将程序的运行环境切换到指定的 GPU 设备上。当你发现你用方式一没有办法使用你指定的GPU，可以用方式二。表示程序只能看到编号为 0 和 1 的 GPU 设备。将程序的运行环境切换到编号为 0 的 GPU 设备上。
torch._C._cuda_setDevice(device) ^^^^^^^^^^^^^^^^^^^^^^^^ AttributeError: module ‘torch._C‘ has no a
weixin_44980011的博客
03-25 495
pycharm报错：torch._C._cuda_setDevice(device) ^^^^^^^^^^^^^^^^^^^^^^^^ AttributeError: module 'torch._C' has no attribute '_cuda_setDevice'网上的答案主要是说：attributeerror: module 'torch._c' has no attribute '_cuda_setdevice'的错误提示可能与PyTorch的CUDA版本有关。换成自己的环境就好了[捂脸]
torch.cuda常用指令
qq_43307074的博客
11-03 8803
torch.cuda常用指令
rt-thread源码分析篇十：rt_console_set_device函数分析
OnlyLove_的博客
08-31 1455
一、前言 rt_console_set_device被rt_hw_board_init调用。 #define RT_CONSOLE_DEVICE_NAME "uart1" #ifdef RT_USING_CONSOLE rt_console_set_device(RT_CONSOLE_DEVICE_NAME); #endif 二、rt_console_set_device函数分析 rt_device_t rt_console_set_device(const char *name) {
【终于解决attributeerror: module ‘torch._c‘ has no attribute ‘_cuda_setdevice‘（多种方案）】
weixin_45914748的博客
04-29 4759
这通常意味着：你使用的 Pytorch 版本中不存在这个特定的底层C API（即CUDA版本问题），或者你可能错误地使用了Pytorch的API（调用错误）。不是gpu版本，就在python命令后面加上 --gpu_ids -1（gpu_ids原来默认值是0），问题解决。最后可以试试把torch_c改为torch。
torch._C._cuda_setDevice(device) RuntimeError: CUDA error: invalid device ordinal
02-25
torch._C._cuda_setDevice(device)是PyTorch中的一个函数，用于设置当前使用的CUDA设备。在这个函数中，device参数表示要设置的设备的索引号。 然而，当出现RuntimeError: CUDA error: invalid device ordinal错误时，意味着指定的设备索引号是无效的。这可能是由于以下几个原因导致的： 1. 设备索引号超出了可用设备的范围：通常情况下，设备索引号从0开始，一直到可用设备的数量减1。如果指定的设备索引号超过了可用设备的数量，就会出现该错误。 2. 没有可用的CUDA设备：如果系统中没有安装或配置CUDA设备，就无法使用CUDA相关功能。在这种情况下，尝试使用CUDA相关函数会导致该错误。 为了解决这个问题，你可以采取以下步骤： 1. 确保设备索引号是有效的：检查你指定的设备索引号是否在可用设备的范围内。你可以使用torch.cuda.device_count()函数来获取可用设备的数量，并确保指定的设备索引号小于该数量。 2. 确保CUDA设备可用：确保你的系统中已正确安装和配置了CUDA设备。你可以通过运行nvidia-smi命令来检查CUDA设备是否可用，并确保驱动程序和CUDA工具包已正确安装。 如果你仍然遇到问题，可以提供更多关于你的环境和代码的详细信息，以便更好地帮助你解决问题。
评论 2
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

当前余额3.43元 前往充值 >
需支付：10.00元
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

Wanderer001

ROIAlign原理

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付：¥1
获取中
扫码支付

您的余额不足，请更换扫码支付或充值

打赏作者

实付元
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明：

1.余额是钱包充值的虚拟货币，按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载，可以购买VIP、付费专栏及课程。

余额充值