【Pytorch的坑】OSError: [WinError 126] 找不到指定的模块caffe2_detectron_ops_gpu.dll” or one of its dependencies

错误描述：

OSError: [WinError 126] 找不到指定的模块。Error loading “D:\Anaconda3\envs\MyCode\Lib\site-packages\torch\lib\caffe2_detectron_ops_gpu.dll” or one of its dependencies

思路描述：

查阅了csdn很多回答，也安装了Microsoft Visual C++ Redistributable，依旧未能解决。这个error提示的是找不到caffe2_detectron_ops_gpu.dll，这个文件尾缀前面加了gpu，我在想也许是之前未安装cudatoolkit的缘故（因为安装cudatoolkit的网络实在太慢，我手动安装了cuda和cudnn）。于是，我决定再用一下官网给的命令，重新把缺少的cudatoolkit和其他包安装一下。

conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch

执行完这个命令后，解决。

测试代码：

import torch
import numpy as np

np_data = np.arange(6).reshape((2, 3))
torch_data = torch.from_numpy(np_data)
tensor2array = torch_data.numpy()
print(
    '\nnumpy array:', np_data,          # [[0 1 2], [3 4 5]]
    '\ntorch tensor:', torch_data,      #  0  1  2 \n 3  4  5    [torch.LongTensor of size 2x3]
    '\ntensor to array:', tensor2array, # [[0 1 2], [3 4 5]]
)

测试结果：

D:\Anaconda3\envs\MyCode\python.exe F:/7.毕业设计/MyCode/test.py

numpy array: [[0 1 2]
 [3 4 5]] 
torch tensor: tensor([[0, 1, 2],
        [3, 4, 5]], dtype=torch.int32) 
tensor to array: [[0 1 2]
 [3 4 5]]

Process finished with exit code 0

深度学习环境：

python版本 3.6.13
pytorch版本 1.8.0
cuda版本 10.2
cudnn版本 7.6.5
torchvision版本 0.9.0