您的位置:首页 > 教育 > 锐评 > 手机软件商店_网页怎么写_保定网站建设方案优化_站长工具樱花

手机软件商店_网页怎么写_保定网站建设方案优化_站长工具樱花

2024/10/5 20:28:36 来源:https://blog.csdn.net/qq_31239371/article/details/142595373  浏览:    关键词:手机软件商店_网页怎么写_保定网站建设方案优化_站长工具樱花
手机软件商店_网页怎么写_保定网站建设方案优化_站长工具樱花

记录一下出现上述问题的一个原因:

pytorch所依赖的cuda版本不满足显卡(GPU)的算力要求

举例来说,显卡是3090,并按照以下命令安装Pytorch:

conda install pytorch==1.7.0 torchvision==0.8.0 torchaudio==0.7.0 cudatoolkit=10.1 -c pytorch

可以看到,这里的cuda是10.1。通过查看算力表可以看到,3090对应的算力是8.6(sm_86)

而不同的cuda所能支持的算力如下:

明显可以看到,cuda10.1并不支持sm_86,需要至少cuda11.1及以上版本才行

因此,安装pytorch时cuda版本一定要大于等于11.1 ,例如:

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

 当然,这里只是满足了cuda的要求,具体安装哪个版本pytorch,还要看对应的python版本等其他限制,此处不再赘述~

 PS:关于cuda对应的算力表,也可以看NVIDIA Datacenter Drivers :: NVIDIA Data Center GPU Driver Documentation

Pytorch历史版本查询:

Previous PyTorch Versions | PyTorch

参考:

[解决bug]CUDA error: no kernel image is available for execution on the device-CSDN博客

【已解决】RuntimeError: CuDA error: no kernel image is available for execution on the device-CSDN博客

pytorch 中注意cuda版本和gpu算力匹配_pytorch对cuda和算力的要求-CSDN博客

显卡算力、驱动版本、CUDA、pytorch之间的关系_显卡驱动是不是cuda版本-CSDN博客

【CUDA】GPU 算力与 CUDA 版本对应关系_cuda算力-CSDN博客

版权声明:

本网仅为发布的内容提供存储空间,不对发表、转载的内容提供任何形式的保证。凡本网注明“来源:XXX网络”的作品,均转载自其它媒体,著作权归作者所有,商业转载请联系作者获得授权,非商业转载请注明出处。

我们尊重并感谢每一位作者,均已注明文章来源和作者。如因作品内容、版权或其它问题,请及时与我们联系,联系邮箱:809451989@qq.com,投稿邮箱:809451989@qq.com