单项选择题

查看gpu显卡数量的函数是（）

A.tensor_gpu.is_cuda
B.torch.cuda.device_count（）
C.tensor_gpu.device
D.torch.cuda.is_available（）

你可能感兴趣的试题

单项选择题

torch.nn.Conv2d中输入通道的参数为（）

A.stride
B.kerner_size
C.in_channels
D.out_channels

单项选择题

提升管道性能描述错误的是（）

A.使用prefetch方法让数据准备和参数迭代两个过程相互并行
B.使用map时设置num_parallel_calls让数据转换过程多进程执行
C.使用cache方法让数据在第一个epoch后缓存到内存中，仅限于数据集不大情形
D.使用map转换时，不用batch，采用向量化的转换方法对每个batch进行转换