为什么我在colab中使用cpu而不是cuda运行脚本

Question

为什么colab对所有检查输出都打印false，即使是在第一个命令中。

如果我不运行这个脚本，colab会为cuda.is_available()返回True。原因是什么？

if __name__ == "__main__":
print(torch.cuda.is_available())
parser = argparse.ArgumentParser('parser')
parser.add_argument("--device", type=int, required=True)
parser.add_argument("--ckptdeblurGAN", type=str, default="./deblurGANLib/pretrained_weights/WGAN_GP/checkpoint-epoch300.pth")
parser.add_argument("--video_root", type=str, default="./imgAndVideos/video/")
parser.add_argument("--cfg", type=str, default="cfg/my_yolov3.cfg")
parser.add_argument("--weights",
type=str,
default="./weights/yolov3spp-59_deblurGAN.pt")
parser.add_argument("--img_size", type=int, default=512)
parser.add_argument(
"--deep_checkpoint",
type=str,
default="deep_sort/deep/checkpoint/resnet50_best_centerloss_final.pt")
# hyperparameter
parser.add_argument("--max_dist", type=float, default=0.2)
# presentation
parser.add_argument("--display", dest="display", action="store_true", default=False)
parser.add_argument("--display_width", type=int, default=1280)
parser.add_argument("--display_height", type=int, default=720)
args = parser.parse_args()
print(torch.cuda.is_available())
if torch.cuda.is_available() and args.device >= 0:

device = torch.device(f'cuda:{args.device}')
torch.cuda.manual_seed(1234)
else:
device = torch.device('cpu')
torch.manual_seed(1234)

print("check!",device)

Answer 1

是否检查了运行时设置：

运行时->修改运行时类型->选择GPU

链接：https://www.tutorialspoint.com/google_colab/google_colab_using_free_gpu.htm