admin 管理员组

文章数量: 1087675

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，安装完毕

本文标签： 系统 Torch Windows False isavailable

版权声明：本文标题：Windows系统下torch.cuda.is_available()返回为False的问题解决 内容由网友自发贡献，该文观点仅代表作者本人， 转载请联系作者并注明出处：http://roclinux.cn/b/1737886263a1891840.html， 本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。如发现本站有涉嫌抄袭侵权/违法违规的内容，一经查实，本站将立刻删除。