从40系显卡换5070TI后CUDA出现问题,解决方法

        因为我这几天换了新的显卡,所以我重新去跑了一下我的代码,然后发现出现下述问题:

RuntimeError: CUDA error: no kernel image is available for execution on the device,我当时感觉就是没有装最新的驱动,然后我在官网装了最新的12.8的CUDATOOLKIT,然后在继续跑代码,发现还是错误,因为50系显卡的框架,是全新的框架,所以老版的pytorch并不能满足,导致不匹配,光官网的pytorch才2.6,所以需要下载下述这个版本:

不建议直接下载,因为太慢了。

建议直接复制这个网址download.pytorch.org/whl/nightly/torch/,在网址中找到下述字眼,

因为下载最新的最好,因为我们的是不稳定版本,就怕后面出现问题就完蛋了。

进入一会等待区,下载完以后就可以看到这个文件了

然后进行你创立的虚拟环境,在进入之前,你先按住shift键点击这个文件,然后就可以复制文件地址,然后输出下列命令行,耐心等待就可以了。

下载完成以后测试一下,是否安装成功,下面是我测试的步骤,大家可以看一下。

在下述步骤之前,可能会出现一个问题,就是你没有numpy这个包,你只需要pip install一下就好了。

到这里就完美结束了。

评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值