虚拟环境装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 中国大陆许可协议进行许可。

posted @   心比天高xzh  阅读(43)  评论(0编辑  收藏  举报
点击右上角即可分享
微信分享提示
评论
收藏
关注
推荐
深色
回顶
收起