指定GPU无效

2024-01-07 17:25:43

今天在进行模型训练的时候遇到这样一个问题：我想要模型在第1块显卡上进行训练，但是不论怎么指定GPU，模型始终是在第0块显卡上进行训练，很是不理解为什么这样？

经过查找资料，发现一个问题：如果在torch之后再指定GPU,那么这个指定是无效的，究竟为什么我还是没有想通，如果有知道原因的小伙伴麻烦下面评论告诉我一下，谢谢！

正确代码如图所示：

import os
os.environ["CUDA_VISIBLE_DEVICES"] = '1'

import torch

文章来源:https://blog.csdn.net/Kelly_Ai_Bai/article/details/135440219
本文来自互联网用户投稿，该文观点仅代表作者本人，不代表本站立场。本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。