设置CUDA_VISIBLE_DEVICES的方法

1.通过bash脚本设置CUDA_VISIBLE_DEVICES

CUDA_VISIBLE_DEVICES=0 nohup python -u train.py

2.通过os.environ来设置CUDA_VISIBLE_DEVICES

你可能感兴趣的:(PyTorch,linux,服务器)