直接换pytorch版本
官方推荐的cuda版本为10.2和11.3,这两种 cuda 支持大多数的 pytorch 版本。
官网
https://pytorch.org/get-started/previous-versions/
直接拷代码下载
PyTorch报CUDA error: no kernel image is available for execution on the device 错误
最新推荐文章于 2024-04-28 17:10:43 发布
直接换pytorch版本
官方推荐的cuda版本为10.2和11.3,这两种 cuda 支持大多数的 pytorch 版本。
官网
https://pytorch.org/get-started/previous-versions/
直接拷代码下载