我正在尝试安装 torch-gpu。我已经安装了 cuda 11.1 并通过命令安装了 torch 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
这太愚蠢了,但在导入过程中我有以下错误 OSError: [WinError 127] Error loading "C:\Users\User\AppData\Roaming\Python\Python38\site-packages\torch\lib\c10_cuda.dll"或其依赖项之一。
。
那个目录下有c10_cuda.dll,我弄了好几天还是没解决。我需要 torch-gpu,这就是使用这种可能性的意义所在,请不要评论“安装 torch-cpu”。
最佳答案
自己回答问题
要安装 torch-gpu,应该:
检查最新(或不是)版本和兼容的 cuda 版本:https://pytorch.org/get-started/locally/ ,此时为11.1。
下载并安装1)步骤中建立版本的cuda(https://developer.nvidia.com/cuda-toolkit-archive)。这取决于 gpu,如果有任何问题,您可以下载旧版本或者安装它?然后您可以在命令行中手动检查版本:
nvcc --version
。使用步骤 1) 中的命令安装 torch(例如
conda install pytorch torchvision torchaudio cudatoolkit=11.1 -c pytorch -c conda-forge
)通过
import torch torch.cuda.is_available()
检查安装是否成功。
在我的例子中,我必须删除之前安装的 torch pip uninstall torch
并使用 conda clean -a
来成功完成安装。
关于installation - 导入 torch OSError : [WinError 127],我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/67122586/