一、背景

 

第一次根据pytorch官网给出的命令,结合清华镜像源安装:

pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu116 -i https://pypi.tuna.tsinghua.edu.cn/simple

安装过程挺快,看到了torch、torchvision、torchaudio的安装过程,但从始至终都没看到含有cuda字眼的安装代码,心生疑惑,果不其然:

二、解决方案

至于怎么找出这个原因的不再赘述,直接上结论:清华镜像源中并没有cu116这个版本(python3.6环境)。

清华大学开源软件镜像站

发现有cuda11.1和cuda11.3,于是卸载之前安装的版本:

pip3 uninstall torch torchvision torchaudio

将安装代码中的116修改为113:

pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu113 -i https://pypi.tuna.tsinghua.edu.cn/simple

此时,安装过程中出现了cu113,预感要成功了。

 果然,问题成功解决。

三、备注

不一定所有torch.cuda.is_available()为false都是这个原因,这是作为新手的我第一次安装pytorch遇到的问题,希望能够帮助到广大铁子!

Logo

腾讯云面向开发者汇聚海量精品云计算使用和开发经验,营造开放的云计算技术生态圈。

更多推荐