欢迎光临散文网 会员登陆 & 注册

Pytorch踩坑实录

2023-04-06 21:02 作者:许许多多的祭日  | 我要投稿

>>> import torch

Traceback (most recent call last):

  File "<stdin>", line 1, in <module>

  File "C:\programming\anaconda\envs\d2l\lib\site-packages\torch\__init__.py", line 122, in <module>

    raise err

OSError: [WinError 127] 找不到指定的程序。 Error loading "C:\programming\anaconda\envs\d2l\lib\site-packages\torch\lib\torch_cuda_cpp.dll" or one of its dependencies.

解决上述问题:

1. CUDA版本查询:支持的版本与Nvidia驱动有关

nvidia-smi

    nvcc-v

2. torch环境冲突

conda uninstall pytorch

pip uninstall pytorch

pip uninstall torchvision

pip uninstall torchaudio

3. 卸载后

conda clean -a

4. 重新安装pytorch

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

5.问题解决


Pytorch踩坑实录的评论 (共 条)

分享到微博请遵守国家法律