【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
·
错误描述:
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
更多推荐
已为社区贡献1条内容
所有评论(0)