jupyter cuda设置

发表于 | 阅读次数: 
1
2
3
4
5
6
import os
os.environ["CUDA_VISIBLE_DEVICES"]="0"

torch.backends.cudnn.enabled = True
print(torch.cuda.device_count())
gpus = [0]
请作者喝一杯咖啡☕️