解决 ZeroDivisionError integer division or modulo by zero

解决 ZeroDivisionError: integer division or modulo by zero

问题

Traceback (most recent call last):
  File "F:/python_projects/BLINK/elq/biencoder/train_biencoder.py", line 593, in <module>
    main(params)
  File "F:/python_projects/BLINK/elq/biencoder/train_biencoder.py", line 248, in main
    valid_subset = len(valid_samples) - len(valid_samples) % torch.cuda.device_count()
ZeroDivisionError: integer division or modulo by zero

原因

下面这条语句的结果为0导致出现问题

torch.cuda.device_count()    # 0

解决方法

可以看到当前安装的torch为CPU版本，而此处的cuda是要使用gpu版本

import torch
print(torch.__version__)
# 1.7.1+cpu
print(torch.cuda.device_count())
# 0

安装gpu版本的torch

pytorch官网安装对应的版本

下面我安装的是conda10.1版本的torch（1.7.1），输出如下：

import torch
print(torch.__version__)
# 1.7.1+cu101
print(torch.cuda.device_count())
# 1

可以使用以下命令查看当前的torch在cuda上是否能用：

import torch
print(torch.cuda.is_available())
# True

重新运行程序，成功解决问题！