No module named 'torch._custom_ops'

[jeffreyrobeson]

(ootd) E:\software\OOTDiffusion\run>python run_ootd.py --model_path E:\software\OOTDiffusion\run\examples\model\model_1.png --cloth_path E:\software\OOTDiffusion\run\examples\garment\02305_00.jpg --scale 2.0 --sample 4 Traceback (most recent call last): File "E:\software\OOTDiffusion\run\run_ootd.py", line 9, in from preprocess.openpose.run_openpose import OpenPose File "E:\software\OOTDiffusion\preprocess\openpose\run_openpose.py", line 20, in from preprocess.openpose.annotator.openpose import OpenposeDetector File "E:\software\OOTDiffusion\preprocess\openpose\annotator\openpose_init_.py", line 15, in from .body import Body File "E:\software\OOTDiffusion\preprocess\openpose\annotator\openpose\body.py", line 14, in from torchvision import transforms File "C:\Users\Administrator\pinokio\bin\miniconda\envs\ootd\lib\site-packages\torchvision_init_.py", line 6, in from torchvision import _meta_registrations, datasets, io, models, ops, transforms, utils File "C:\Users\Administrator\pinokio\bin\miniconda\envs\ootd\lib\site-packages\torchvision_meta_registrations.py", line 4, in import torch._custom_ops ModuleNotFoundError: No module named 'torch._custom_ops'

(ootd) E:\software\OOTDiffusion\run>

[jeffreyrobeson]

How can I solve this problem?

[jeffreyrobeson]

pip uninstall torch pip uninstall torchvision conda clean -a pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu117 conda install pytorch torchvision torchaudio pytorch-cuda=11.7 -c pytorch -c nvidia python -c "import torch; print(torch.cuda.is_available())" nvcc --version python -m torch.utils.collect_env

[jeffreyrobeson]

The problem has been solved

[508135669]

I want to know that “ No module named 'torch._custom_ops' ” this problem,how should be solved

[chilljudaoren]

pip uninstall torch pip uninstall torchvision conda clean -a pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu117 conda install pytorch torchvision torchaudio pytorch-cuda=11.7 -c pytorch -c nvidia python -c "import torch; print(torch.cuda.is_available())" nvcc --version python -m torch.utils.collect_env

May I ask if cuda=11.6 is OK?

[chilljudaoren]

I want to know that “ No module named 'torch._custom_ops' ” this problem,how should be solved

You can follow the author's method above.