下载pytorch,import torch显示为false,一篇解决

在下载的时候,踩了很多坑才弄好,所以来通俗易懂地记录一下错误,希望能对其他人有帮助。

首先要下载好cuda和cudnn,直接从官网选择对应的版本,获取相关的命名在anaconda中运行。

记得要加上后面的c pytorch,否则就会默认下载为cpu版本

下载pytorch,import torch显示为false,一篇解决_第1张图片

如果已经下载为cpu版本,必须得先卸载再重新下。

可以输入命令conda uninstall pytorch-mutex进行卸载,普通的卸载无法全部卸载cpuonly模块。

接着就可以重新安装了,网上还提供了一种方法,先把对应的pytorch安装包下载下来()

再用命令conda install pytorch torchvision cudatoolkit=10.0 -c pytorch安装

好不容易以为安装好了,结果又出现了OSError 找不到模块,查了说是要把官网的命令再运行一遍,补充没有的模块。

conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch

ps:如果是自己的cuda版本太新了,pytorch上面没有,安一个最新版本的就可以了。比如我的cuda版本是11.4,在安装时就选择的是cuda11.3对应的pytorch。

安装软件要么一次成功,要么就得花好久解决错误,慢慢一个一个方法试就可以了。

祝大家都能成功安装!还有什么问题也可以在评论区询问,说不定我也遇到过哈哈哈哈

你可能感兴趣的:(pytorch,人工智能,python)