如何使用特定显卡跑pyTorch
如何使用特定显卡跑pyTorch。 方法一：改变系统变量使得仅目标显卡可见：
export CUDA_VISIBLE_DEVICES=0 #这里是要使用的GPU编号，正常的话是从0开始然后再运行pyTorch即可。 方法二：使用torch.cuda接口：
#在生成网络对象之前：
torch.cuda.set_device(0)方法三：使用多pytorch的并行GPU接口：
net = torch.nn.DataParallel(model, device_ids=[0])方法四：初始化模型时
net = Net.cuda(0)本文参与 腾讯云自媒体同步曝光计划，分享自作者个人站点/博客。
原始发表：2020/01/14 ，如有侵权请联系 cloudcommunity@tencent.com 删除
评论
登录后参与评论
推荐阅读
相关产品与服务
GPU 云服务器
GPU 云服务器（Cloud GPU Service，GPU）是提供 GPU 算力的弹性计算服务，具有超强的并行计算能力，作为 IaaS 层的尖兵利器，服务于生成式AI，自动驾驶，深度学习训练、科学计算、图形图像处理、视频编解码等场景。腾讯云随时提供触手可得的算力，有效缓解您的计算压力，提升业务效率与竞争力。
腾讯云开发者
