os.environ[‘CUDA_VISIBLE_DEVICES‘]指定GPU无法生效解决方案

解决方法一：
将语句放在 import torch 之前

os.environ['CUDA_VISIBLE_DEVICES'] = args.gpu_devices
import torch

解决方法二：
命令行运行时指定

CUDA_VISIBLE_DEVICES=x python  xxx.py
版权声明：本文为weixin_44229976原创文章，遵循CC 4.0 BY-SA版权协议，转载请附上原文出处链接和本声明。