大佬教程收集整理的这篇文章主要介绍了安装 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.2
和 11.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,请注明来意。