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

你可能感兴趣的:(深度学习,人工智能,python,pytorch)