指定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
本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。