导入火炬 OSError: [WinError 127]

Question

我正在尝试安装 torch-gpu。我已经安装了 cuda 11.1 并通过命令 pip3 install torch==1.8.1+cu111 torchvision==0.9.1+cu111 torchaudio===0.8.1 -f https://download.pytorch.org/whl/torch_stable.html

安装了 Torch

这太愚蠢了，但在导入过程中出现以下错误 OSError: [WinError 127] Error loading "C:\Users\User\AppData\Roaming\Python\Python38\site-packages\torch\lib\c10_cuda.dll" or one of its dependencies.。

c10_cuda.dll在那个目录下，我用了几天还是解决不了。我需要 torch-gpu，这就是使用这种可能性的意义所在，请不要评论“安装 torch-cpu”。

Answer 1

自己回答问题

要安装 torch-gpu 应该：

1. 检查最新（或不是）版本和兼容的 cuda 版本：https://pytorch.org/get-started/locally/，目前是 11.1。

2. 在 1) 步骤 (https://developer.nvidia.com/cuda-toolkit-archive) 中下载并安装已建立版本的 cuda。这取决于gpu，如果有任何问题，您可以下载旧版本还是安装了它？然后您可以在命令行中手动检查版本：nvcc --version。

3. 使用在 1) 步骤中找到的命令安装火炬（例如 conda install pytorch torchvision torchaudio cudatoolkit=11.1 -c pytorch -c conda-forge）

4. 检查安装成功为 import torch torch.cuda.is_available()。

就我而言，我必须删除之前安装的 Torch pip uninstall torch 并使用 conda clean -a 才能成功结束安装。