我已使用命令安装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_nvrtc.dll。可能有什么错误以及如何修复此错误
你的机器上有GPU吗?尝试在没有CUDA的情况下安装,这为我修复了它
我也面临同样的问题。如果您的操作系统是Windows,那么我建议您使用Anaconda并在单独的conda环境中安装pytorch。快速解决方案是在谷歌上搜索nvcuda.dll文件并下载该文件。如果您在Jupyter笔记本上运行代码,输出将为您提供conda环境的“lib”文件夹的完整路径。默认情况下,它是“C:\Users\YourUserName\anaconda3\envs\mera\u beta\Lib\site packages\torch\Lib”。转到此目录并将文件粘贴到此文件夹中。重新运行代码。希望它能运行
我也有同样的问题。当我安装PyTorch纯CPU版本时,它对我有效。使用
conda install pytorch torchvision cpuonly -c pytorch
可以找到here相关问题 更多 >
编程相关推荐