You need to enable JavaScript to run this app.
最新活动
大模型
产品
解决方案
定价
生态与合作
支持与服务
开发者
了解我们

Windows环境下PyTorch安装成功后出现ImportError求助

File "C:\Users\PycharmProjects\chatbot-light\pytorch\rnn_attention\seq2seq_translation_tutorial.py", line 93, in import torch File "C:\Users\AppData\Local\Programs\Python\Python36\lib\site-packages\torch_init_.py", line 78, in from torch._C import * ImportError: DLL load failed: The specified module could not be found

嘿,这个问题我太熟了!这是PyTorch安装后常见的DLL依赖问题,咱们从几个最可能的原因入手解决:

1. 补装必要的VC++运行库

PyTorch底层依赖Microsoft Visual C++ Redistributable for Visual Studio 2019,哪怕你没装Visual Studio,这个运行库也必须要有。直接下载对应x64版本的安装包(你的Python3.6应该是64位的),安装完成后重启电脑,再运行代码试试。

2. 确保PyTorch版本适配Python3.6

Python3.6已经停止维护了,很多新的PyTorch版本不再支持它。你需要安装专门适配Python3.6的PyTorch版本,比如PyTorch 1.12.1是最后一批支持Python3.6的版本之一。

先卸载现有版本:

pip uninstall torch torchvision torchaudio -y

然后根据你的需求安装:

  • 如果是CPU版本:
pip install torch==1.12.1+cpu torchvision==0.13.1+cpu torchaudio==0.12.1 --extra-index-url https://download.pytorch.org/whl/cpu
  • 如果是GPU版本(适配CUDA 11.3):
pip install torch==1.12.1+cu113 torchvision==0.13.1+cu113 torchaudio==0.12.1 --extra-index-url https://download.pytorch.org/whl/cu113

3. 检查Python是否为64位

PyTorch只支持64位Python,如果你装的是32位Python3.6,肯定会出这个问题。打开命令行输入:

python -c "import sys; print(sys.maxsize > 2**32)"

如果输出True就是64位,没问题;如果是False,你得卸载32位Python,重装64位的Python3.6。

4. GPU版本的额外检查(如果用显卡)

如果你装的是带CUDA的PyTorch,要确保你的NVIDIA显卡驱动版本和PyTorch对应的CUDA版本匹配。比如上面的CUDA 11.3需要驱动版本至少465.89以上,更新到符合要求的驱动,或者先换成CPU版本验证是否能正常运行。

内容的提问来源于stack exchange,提问作者tompal18

火山引擎 最新活动