机器上有八张显卡,占用了两三块,我选择了空余的GPU仍报错,
使用下面代码查看可用GPU时,n_gpu为1,一脸问号
n_gpu = torch.cuda.device_count()
使用GPU方法:
import os
os.environ["CUDA_VISIBLE_DEVICES"] = "0"
释放GPU代码:
torch.cuda.empty_cache()
机器上有八张显卡,占用了两三块,我选择了空余的GPU仍报错,
使用下面代码查看可用GPU时,n_gpu为1,一脸问号
n_gpu = torch.cuda.device_count()
使用GPU方法:
import os
os.environ["CUDA_VISIBLE_DEVICES"] = "0"
释放GPU代码:
torch.cuda.empty_cache()