pytorch指定gpu序号不起作用

import os

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

要放在 import torch 的前面

TensorFlow 同理

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