程序问答   发布时间:2022-06-02  发布网站:大佬教程  code.js-code.com
大佬教程收集整理的这篇文章主要介绍了安装 Pytorch cuda 后, torch.cuda.is_available() 显示 false 。该怎么办?大佬教程大佬觉得挺不错的,现在分享给大家,也给大家做个参考。

如何解决安装 Pytorch cuda 后, torch.cuda.is_available() 显示 false 。该怎么办??

开发过程中遇到安装 Pytorch cuda 后, torch.cuda.is_available() 显示 false 。该怎么办?的问题如何解决?下面主要结合日常开发的经验,给出你关于安装 Pytorch cuda 后, torch.cuda.is_available() 显示 false 。该怎么办?的解决方法建议,希望对你解决安装 Pytorch cuda 后, torch.cuda.is_available() 显示 false 。该怎么办?有所启发或帮助;

我已经通过这个安装了 pytorch cuda ..

conda install pytorch torchvision torchaudio cudatoolkit=11.1 -c pytorch -c nvIDia

我的 cuda 版本是 11.2 。我正在使用 windows 10 。 Pytorch cuda 11.2 目前不可用。 (pytorch.org) 所以我安装了 11.1 版本。

(使用 nvIDia-smi) Cuda version

但显示为假。

torch.cuda.is_available() >>> false

10.211.1 版本我都试过了。

据我所知,我不需要为 pytorch 安装 cuda 工具包

解决方法

您不应将软件包安装到您的基础环境中。使用必要的工具创建一个单独的环境。

例:使用 Python v3.7 和火炬包创建名为 dlearn 的环境

conda create -n dlearn python=3.7 pytorch torchvision torchaudio cudatoolkit=11.1 -c pytorch -c nvidia

激活并使用您的 dlearn 环境

conda activate dlearn
python -c "import torch;print(torch.cuda.is_available())"
# this should echo True if all is well

目前支持的 cudatoolkit 是 11.1,它可以与 11.2 驱动程序配合使用。他们迟早会更新它。如果需要,您可以从 GitHub PyTorch 构建以获取最新版本。步骤比上面复杂。

大佬总结

以上是大佬教程为你收集整理的安装 Pytorch cuda 后, torch.cuda.is_available() 显示 false 。该怎么办?全部内容,希望文章能够帮你解决安装 Pytorch cuda 后, torch.cuda.is_available() 显示 false 。该怎么办?所遇到的程序开发问题。

如果觉得大佬教程网站内容还不错,欢迎将大佬教程推荐给程序员好友。

本图文内容来源于网友网络收集整理提供,作为学习参考使用,版权属于原作者。
如您有任何意见或建议可联系处理。小编QQ:384754419,请注明来意。