Torch之CUDA使用指定显卡_数据to cuda指定显卡

作者：小蓝xlanll | 2024-04-14 05:54:23

踩

数据to cuda指定显卡

export CUDA_VISIBLE_DEVICES=0 #这里是要使用的GPU编号，正常的话是从0开始

#在生成网络对象之前：
torch.cuda.set_device(0)

CUDA_VISIBLE_DEVICES=7 python retinanet/main.py

net = torch.nn.DataParallel(model, device_ids=[0])

net = Net.cuda(0)

注：

参考博客：https://blog.csdn.net/weixin_40712763/article/details/82846509

声明：本文内容由网友自发贡献，不代表【wpsshop博客】立场，版权归原作者所有，本站不承担相应法律责任。如您发现有侵权的内容，请联系我们。转载请注明出处：https://www.wpsshop.cn/w/小蓝xlanll/article/detail/420497