cuda10.2不再适用于windows(cuda和cudnn均正常安装但报错torch.cuda.is_available()为False)

        PC:Huaweimatebook14

        驱动:GeForce MX350    

        python:3.9

    太长不看版:windows下cuda10.2安装pytorch gpu版本仅能使用1.10.1及之前版本,代码如下:

conda install pytorch==1.10.1 torchvision==0.11.2 torchaudio==0.10.1 cudatoolkit=10.2 -c pytorch

        最近由于实习需求要在笔记本上安装gpu版本的pytorch(公司给配的台式机还没到只能先在笔记本上实验一下,不得不在我的陈旧笔记本上去跑)。之前都是在服务器上跑代码所以自己机器上的很久没有用了,上次用还是一年前的大作业,之前的环境配的很乱,因此打算新建环境安装pytorch。

        根据网上的指导在Nvidia控制面板中找到支持的cuda版本,我的是10.2,应该是只能安装低于这个版本的cuda。安装好后,去官网上查找之前的安装命令,如下:

         到创建好的虚拟环境中输入命令进行安装(若安装失败则需要进行换源)。本以为一切大功告成,但是当按照传统三部曲去检查是否安装成功时(先在命令行对应环境中输入python再输入以下命令),最后一步却出现了False的错误。

import torch
print(torch.__version__)
print(torch.cuda.is_available())

        开始调转方向去检查cuda和cudnn,按照链接方法http://t.csdn.cn/vrJGK检查后,发现没有问题。于是开始在网上瞎逛,偶然发现有文章说cuda10,2不再支持Windows,pytorch官网说明为(CUDA-10.2 PyTorch builds are no longer available for Windows, please use CUDA-11.3)当然这个说明只在10.2版本还在主页的时候能够看到(也是够坑,不支持为啥还要放链接(抓狂)。可是电脑驱动也装不了更高版本的cuda了呀,难道就不能用了吗?

        不甘心地去尝试更早版本的Pytoch,安装了1.10.1的版本如下,发现是可用的!原来只是后边的版本不支持windows版本的cuda10.2了,但是官网里面却还是给了cuda10.2对应的linux和windows下载命令,有点迷惑住。

# CUDA 10.2
conda install pytorch==1.10.1 torchvision==0.11.2 torchaudio==0.10.1 cudatoolkit=10.2 -c pytorch

# CUDA 11.3
conda install pytorch==1.10.1 torchvision==0.11.2 torchaudio==0.10.1 cudatoolkit=11.3 -c pytorch -c conda-forge

# CPU Only
conda install pytorch==1.10.1 torchvision==0.11.2 torchaudio==0.10.1 cpuonly -c pytorch

抱着严谨认真的态度,又去测试了一下其他版本,发现1.11.0及之后的都不行,所以uu们如果是相同情况只有cuda10.2并且要装pytorchgpu版本的话就使用上方代码就好。

  • 3
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 5
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 5
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值