CUDA error: no kernel image is available for execution
经过各种百度都没解决，后来在官网找到了答案，我把网址放在这里，大家可以自己查看，https://arnon.dk/matching-sm-architectures-arch-and-gencode-for-various-nvidia-cards/截图是解决问题的描述部分，因为我cuda安装的是11.0版本的，命令输入 export TORCH_CUDA_ARCH_LIST="8.0"...
·
经过各种百度都没解决，后来在官网找到了答案，我把网址放在这里，大家可以自己查看，
https://arnon.dk/matching-sm-architectures-arch-and-gencode-for-various-nvidia-cards/
截图是解决问题的描述部分，因为我cuda安装的是11.0版本的，
命令输入 export TORCH_CUDA_ARCH_LIST="8.0"
一站式 AI 云服务平台
更多推荐
0
0- 0
已为社区贡献3条内容
所有评论(0)