UserWarning：针对 CuDNN 问题应用了解决方法，安装 nvrtc.so

Question

我正在下面训练一个卷积神经网络。

def __init__(self, n_channels, n_classes):
    super().__init__()
    self.model = models.mobilenet_v3_large(pretrained=True)
    self.model.classifier[-1] = nn.Linear(1280, n_classes)
    self.model.features[0][0] = nn.Conv2d(
        n_channels,
        16,
        kernel_size=(3, 3),
        stride=(2, 2),
        padding=(1, 1),
        bias=False,
    )

警告中只有另一件事是

返回 F.conv2d(输入, 权重, 偏差, self.stride,

我不知道这个警告是什么以及我能做什么。如果需要有关系统的任何信息，我可以编辑问题并添加到其中。

Answer 1

这似乎是 CUDA 库的问题pytorch 2.0.x+cu118。有几个可能的修复方法。

1. 降级到以前的 CUDA 库 cu117。例如，用点：

pip3 uninstall pytorch
pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu117) 

2. 要继续使用 cu118，请在 torch 包库中创建符号链接。

cd venv/python3.10/site-packages/torch/lib  # (or the path to dist-packages/torch/lib, depending on how you installed pytorch)

ln -s libnvrtc-*.so.11.2 libnvrtc.so

或者，您应该能够忽略该警告；它不应该有任何副作用或影响。

我在这个 GitHub 问题中找到了上述解决方案，以及对该问题的更深入的讨论。