错误描述:

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
Logo

为开发者提供学习成长、分享交流、生态实践、资源工具等服务,帮助开发者快速成长。

更多推荐