admin管理员组文章数量:1026989
Windows系统如何安装cuda-GPU版本
第一步,从英伟达官网下载cuda并安装,下载链接:https://developer.nvidia/cuda-toolkit(注意GPU与cuda的版本对应)
第二步,找到cuda对应的torch版本(网上很多)
注:不要使用pytorch官网上的命令安装,它会帮你安装cpu的版本
第三步,下载torch-gpu版本,下载链接:https://download.pytorch/whl/torch/
(如下图,cu115指cuda11.5,cp310指python3.10,win指windows,amd64指64位操作系统)
第四步,打开cmd,使用pip本地安装whl文件(这里推荐使用conda虚拟环境),如下图:
pip install <下载文件路径/文件名称.whl>
安装成功
安装成功后,使用python的print(torch.cuda.is_available()),结果输出为True,安装完毕
Windows系统如何安装cuda-GPU版本
第一步,从英伟达官网下载cuda并安装,下载链接:https://developer.nvidia/cuda-toolkit(注意GPU与cuda的版本对应)
第二步,找到cuda对应的torch版本(网上很多)
注:不要使用pytorch官网上的命令安装,它会帮你安装cpu的版本
第三步,下载torch-gpu版本,下载链接:https://download.pytorch/whl/torch/
(如下图,cu115指cuda11.5,cp310指python3.10,win指windows,amd64指64位操作系统)
第四步,打开cmd,使用pip本地安装whl文件(这里推荐使用conda虚拟环境),如下图:
pip install <下载文件路径/文件名称.whl>
安装成功
安装成功后,使用python的print(torch.cuda.is_available()),结果输出为True,安装完毕
本文标签: 系统torchWindowsfalseisavailable
版权声明:本文标题:Windows系统下torch.cuda.is_available()返回为False的问题解决 内容由热心网友自发贡献,该文观点仅代表作者本人, 转载请联系作者并注明出处:http://it.en369.cn/jiaocheng/1737886260a1516664.html, 本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌抄袭侵权/违法违规的内容,一经查实,本站将立刻删除。
发表评论