jaketae / deep-malware-detection

A neural approach to malware detection in portable executables
MIT License
66 stars 15 forks source link

Training.py error #13

Open thundersoft90 opened 1 month ago

thundersoft90 commented 1 month ago

Hi,

When I tried to execute train.py I encounter error as follows

D:\x_malwdetc\x_mawdetc\src\deep_malware_detection>python train.py --benign_dir="D:\x_malwdetc\x_mawdetc\src\bin\raw\dll-headersiz" --malware_dir="D:\x_malwdetc\x_mawdetc\src\bin\raw\dasmalwerk-headersiz" Traceback (most recent call last): File "D:\x_malwdetc\x_mawdetc\src\deep_malware_detection\train.py", line 4, in import models File "D:\x_malwdetc\x_mawdetc\src\deep_malware_detection\models.py", line 1, in import torch File "C:\Users\x\AppData\Local\Packages\PythonSoftwareFoundation.Python.3.12_qbz5n2kfra8p0\LocalCache\local-packages\Python312\site-packages\torch__init__.py", line 141, in raise err OSError: [WinError 126] Belirtilen modül bulunamadı. Error loading "C:\Users\x\AppData\Local\Packages\PythonSoftwareFoundation.Python.3.12_qbz5n2kfra8p0\LocalCache\local-packages\Python312\site-packages\torch\lib\shm.dll" or one of its dependencies.

Would you help me to resolve this issue ? Thanks in advance

jaketae commented 1 month ago

The error is not coming from this library or code. Can you double check that you are able to import PyTorch in your virtual environment?

>>> import torch
thundersoft90 commented 1 month ago

By the way, in files uploading to the neural network, which parameter is considered in learning process to determine whether it is benignware or malware ? For instance is it done by PE format or something else ?

jaketae commented 1 month ago

It's the PE format. The README contains some references I used for the implementation.