os.environ[“CUDA_VISIBLE_DEVICES“]学习总结

今天发现一个很有意思的东西

import torch 
import os 

# Specify the GPU device
os.environ["CUDA_VISIBLE_DEVICES"] = "1"
print(torch.cuda.is_available())

但是如果修改下面的设置后，结果就变成了

import torch 
import os 

# Specify the GPU device
os.environ["CUDA_VISIBLE_DEVICES"] = "0"
print(torch.cuda.is_available())