pytorch安装解决torch.cuda.is_available()返回False的问题

最新推荐文章于 2024-05-31 01:06:31 发布

weixin_46332557

最新推荐文章于 2024-05-31 01:06:31 发布

阅读量1.9k

点赞数

文章标签： pytorch 神经网络深度学习

本文链接：https://blog.csdn.net/weixin_46332557/article/details/111997866

版权

在Windows Server 2019上，博主遇到PyTorch无法识别CUDA的问题。原来通过Anaconda安装的PyTorch是CPU版本。解决方法包括卸载现有版本，从PyTorch官网下载对应CUDA版本的whl文件，然后使用pip手动安装。经过这些步骤，torch.cuda.is_available()成功返回True。

摘要由CSDN通过智能技术生成

@windows server 2019 解决torch.cuda.is_available()返回False的问题。
博主最初是通过在Anaconda环境下执行“conda install pytorch”的方法安装的，如图一所示。
在这里插入图片描述
打开anaconda安装目录下的python或者通过cmd进入python后输入“print(torch.version)”能正确显示版本号，输入"torch.cuda.is_available()"后返回False,于是博主开始了漫长的踩坑之旅。最终找到的错误原因和解决方法如下：
**错误原因：**通过anaconda安装的pytorch为CPU版本，而非GPU版本。
解决方法：
1、在Anaconda中卸载已安装的pytorch
在这里插入图片描述
2、去pytorch官网下载所需版本的whl文件。此处可参考博文https://blog.csdn.net/qq_25485287/article/details/100765581
此处补充一点：复制下载链接在浏览器打开后（建议使用迅雷搜索链接），会出现很多下载链接