安装GPU版本Pytorch失败：torch.cuda.is_available()为False

问题所在

检查conda list发现，实际安装的Pytorch为CPU版本（虽然安装时明确指定了cuda版本）：
image
上图中可以看出，Pytorch的描述为：py3.9_cpu_0

解决办法

有可能是因为环境中存在一个叫“cpuonly”的包，导致无法安装GPU版本Pytorch：
image
卸载掉它即可，卸载它时会自动将Pytorch更新为GPU版本：
image
如果检查发现环境中其实并没有这个包呢？可以借鉴此方法，先安装一个“cpuonly”包，再卸载掉它，Pytorch也会自动更新为GPU版本。

conda unistall pytorch
conda istall pytorch
posted @ 2024-01-17 22:14  Yu_tiann  阅读(203)  评论(0)  编辑  收藏  举报
​