首页
学习
活动
专区
工具
TVP
发布
社区首页 >问答首页 >Anaconda / Pytorch -尝试使Pytorch工作时出现错误

Anaconda / Pytorch -尝试使Pytorch工作时出现错误
EN

Stack Overflow用户
提问于 2018-05-06 08:01:27
回答 2查看 1K关注 0票数 1

我在Windows10系统上尝试使用Anaconda安装Pytorch,但每次尝试都会返回各种错误。我从https://anaconda.org/peterjc123/pytorch/files下载了该软件包的windows兼容的tar文件,显然能够使用conda install filename.tar.bz2安装该软件包。显然,我的意思是Pytorch出现在我为使用Pytorch而设置的环境中可用包的列表中。但是当我加载这个包时,我得到了这个错误:

代码语言:javascript
复制
>>> import torch
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
  File "C:\Users\conner\Anaconda3\envs\pytorch\lib\site-packages\torch\__init__.py", line 76, in <module>
    from torch._C import *
ImportError: DLL load failed: The specified module could not be found.

我还尝试用conda update --all更新蟒蛇,这引发了一个相关的警告:

代码语言:javascript
复制
(pytorch) C:\Users\User 1\Downloads>conda update --all
Fetching package metadata .............
Solving package specifications:
Warning: ['Dependency missing in current win-64 channels: \n  - pytorch -> mkl >=2018'], skipping
Fetching package metadata .............
Solving package specifications:
Warning: ['Dependency missing in current win-64 channels: \n  - pytorch -> mkl >=2018'], skipping

NoPackagesFoundError: Dependency missing in current win-64 channels:
  - pytorch -> mkl >=2018

mkl包还会出现在连接到pytorch环境的包列表中。我将这个错误解释为没有足够新的mkl版本。这是正确的吗?还有其他的见解或建议吗?我真的需要安装pytorch。

EN

回答 2

Stack Overflow用户

发布于 2019-01-28 01:00:43

我建议您创建一个new conda enviroment,并尝试以这种方式重新安装PyTorch:

要通过Anaconda安装PyTorch,并且没有支持CUDA的conda系统或不需要CUDA,请使用以下conda命令。

conda install pytorch-cpu torchvision-cpu -c pytorch

要通过Anaconda安装PyTorch,并且您正在使用CUDA9.0,请使用以下conda命令:

conda install pytorch torchvision -c pytorch

CUDA 8.x

conda install pytorch torchvision cuda80 -c pytorch

CUDA 10.0

conda install pytorch torchvision cuda100 -c pytorch

#Inno

票数 1
EN

Stack Overflow用户

发布于 2020-08-04 22:46:13

根据我的经验,原因是Pytorch不能调用你的系统推荐的CUDA dll文件,所以你必须使用确切的CUDA版本的dll来安装Pytorch。

要通过Anaconda安装Pytorch并正常工作,您应该为您的硬件选择合适的CUDA版本。例如,

CUDA 9.0

conda install pytorch==1.0.1 torchvision==0.2.2 cudatoolkit=9.0 -c pytorch

CUDA 10.0

conda install pytorch==1.0.1 torchvision==0.2.2 cudatoolkit=10.0 -c pytorch

仅CPU

conda install pytorch-cpu==1.0.1 torchvision-cpu==0.2.2 cpuonly -c pytorch

如果正确安装了CUDA和CUDA工具包驱动程序,则无需手动选择CUDA版本。

我希望它能对你有所帮助。

票数 0
EN
页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接:

https://stackoverflow.com/questions/50195080

复制
相关文章

相似问题

领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档