jupyter cuda设置

Posted on Views: Views: 
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]