虚拟环境装torch与cuda
遇到问题1
在python环境中导入torchvision的时候,出现了以下错误
ImportError: cannot import name 'PILLOW_VERSION' from 'PIL'
问题:Pillow包版本过高。
解决方法:1.卸载新版本 pip uninstall Pillow
2.安装新版本 pip install Pillow==6.2.2
备注:通过conda进行uninstall好像会把torchvision一起卸掉
遇到问题2
报错Torch not compiled with CUDA enabled
但nvcc -V
是有输出的
问题:Torch找不到cuda,版本不太对
解决方法:1.卸载 pip uninstall torch torchvision torchaudio
2. cuda版本11.8
pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118
正确截图:
错误截图:
本文作者:心比天高xzh
本文链接:https://www.cnblogs.com/xzh-personal-issue/p/18102732
版权声明:本作品采用知识共享署名-非商业性使用-禁止演绎 2.5 中国大陆许可协议进行许可。
【推荐】国内首个AI IDE,深度理解中文开发场景,立即下载体验Trae
【推荐】编程新体验,更懂你的AI,立即体验豆包MarsCode编程助手
【推荐】抖音旗下AI助手豆包,你的智能百科全书,全免费不限次数
【推荐】轻量又高性能的 SSH 工具 IShell:AI 加持,快人一步