因为我这几天换了新的显卡,所以我重新去跑了一下我的代码,然后发现出现下述问题:
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一下就好了。
到这里就完美结束了。