Pytorch не смог найти модуль

Question

Я установил pytorch с помощью команды:

conda install pytorch torchvision cudatoolkit=10.2 -c pytorch -y

Python жалуется на факел импорта строки с сообщением:

Could not find module 'C:\ProgramData\Anaconda3\envs\edx\lib\site-packages\torch\lib\caffe2_nvrtc.dll' (or one of its dependencies). Try using the full path with constructor syntax

Этот каталог содержит библиотеку caffe2_nvrt c .dll. Что может быть не так и как исправить эту ошибку?

Answer 1

Я столкнулся с той же проблемой. Если ваша ОС Windows, я бы рекомендовал использовать Anaconda и установить pytorch в отдельной среде conda. Быстрое решение - поиск файла nvcuda.dll в Google и загрузка этого файла. Если вы запускаете код на ноутбуке Jupyter, в выводе будет указан полный путь к папке lib в среде conda. По умолчанию это 'C: \ Users \ YourUserName \ anaconda3 \ envs \ mera_beta \ Lib \ site-packages \ torch \ lib'. Go в этот каталог и вставьте файл в эту папку. Перезапустите свой код. Надеюсь, он заработает.

Answer 2

У меня была такая же проблема. У меня это сработало, когда я установил версию только для процессора PyTorch. Используйте conda install pytorch torchvision cpuonly -c pytorch, которое можно найти здесь

Answer 3

У вас есть графический процессор на вашем компьютере? Попробуйте установить без CUDA, и это исправило для меня.