安装好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环境