安装好pytorch和cuda后运行程序报错：OSError: [WinError 127] 找不到指定的程序

1 安装报错

从装好pytorch和cuda后运行程序报错：
OSError: [WinError 127] 找不到指定的程序。 Error loading "D:\soft\anaconda\envs\pytorch\Lib\site-packages\torch\libcaffe2_detectron_ops.dll’ ,or one of its dependencies
这是因为安装的pytorch版本和cuda版本不匹配。卸载后在官网重新下载相匹配的对应的版本，安装就可以了。
官网：pytorch
进入到建好的虚拟环境：

activeta pytorch

安装：

conda install pytorch==1.7.1 torchvision==0.8.2 torchaudio==0.7.2 cudatoolkit=10.1 -c pytorch

这里我的电脑cuda版本为10.1，所以安装对应的pytorch1.7.1
当然对于不同的电脑cuda版本，需要安装不同版本的pytorch和torchvision
官网也给出了很多相匹配的版本：

# CUDA 9.2
conda install pytorch==1.7.1 torchvision==0.8.2 torchaudio==0.7.2 cudatoolkit=9.2 -c pytorch# CUDA 10.1
conda install pytorch==1.7.1 torchvision==0.8.2 torchaudio==0.7.2 cudatoolkit=10.1 -c pytorch# CUDA 10.2
conda install pytorch==1.7.1 torchvision==0.8.2 torchaudio==0.7.2 cudatoolkit=10.2 -c pytorch# CUDA 11.0
conda install pytorch==1.7.1 torchvision==0.8.2 torchaudio==0.7.2 cudatoolkit=11.0 -c pytorch# CPU Only
conda install pytorch==1.7.1 torchvision==0.8.2 torchaudio==0.7.2 cpuonly -c pytorch

对应安装就可以了。

2验证pytorch是否安装成功

切换到对应环境并按照如下步骤输入：

得到结果：

说明安装成功。
验证完成之后，可以quit()保存退出。
参考博客：如何搭建pytorch环境

本文来自互联网用户投稿，文章观点仅代表作者本人，不代表本站立场，不承担相关法律责任。如若转载，请注明出处。 如若内容造成侵权/违法违规/事实不符，请点击【内容举报】进行投诉反馈！